Update app.py

app.py

@@ -203,6 +203,8 @@ def main():
 
     progress=gr.Progress()
 
+    adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
+
     repo="stabilityai/sd-vae-ft-mse-original"
     ckpt="vae-ft-mse-840000-ema-pruned.safetensors"
     vae = load_file(hf_hub_download(repo, ckpt), device=device)
@@ -211,9 +213,7 @@ def main():
     ckpt=f"sdxl_lightning_{step}step_unet.safetensors"
     unet = load_file(hf_hub_download(repo, ckpt), device=device)
 
-    repo = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
-    ckpt="feature_extractor/preprocessor_config.json"
-    fext = load_file(hf_hub_download(repo, ckpt), device=device)
+    #repo = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
     
     #repo = "emilianJR/epiCRealism"
     #ckpt = "unet/diffusion_pytorch_model.safetensors"
@@ -223,7 +223,7 @@ def main():
     ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
     base = "black-forest-labs/FLUX.1-schnell"
     
-    pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, feature_extractor=fext, unet=unet, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
+    pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, feature_extractor=None, image_encoder=None, unet=unet, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
     pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
     pipe.enable_free_init(method="butterworth", use_fast_sampling=False)