Update app.py

app.py

@@ -9,14 +9,14 @@ import gradio as gr
 import numpy as np
 from lxml.html import fromstring
 from pathos.threading import ThreadPool as Pool
-from diffusers import AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler
+from diffusers import AnimateDiffPipeline, MotionAdapter, DDIMScheduler
 from diffusers.pipelines.flux import FluxPipeline
 from diffusers.utils import export_to_gif
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-dtype = torch.bfloat16
+dtype = torch.float16
 
 step = 4
 repo = "ByteDance/AnimateDiff-Lightning"
@@ -27,7 +27,20 @@ base = "black-forest-labs/FLUX.1-dev"
 adapter = MotionAdapter().to(device, dtype)
 adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
 pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
-pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
+scheduler = DDIMScheduler.from_pretrained(
+    base,
+    subfolder="scheduler",
+    clip_sample=False,
+    timestep_spacing="linspace",
+    beta_schedule="linear",
+    steps_offset=1,
+)
+pipe.scheduler = scheduler
+pipe.enable_vae_slicing()
+pipe.enable_model_cpu_offload()
+
+
+
 
 def translate(text,lang):