Update app.py

app.py

@@ -120,7 +120,7 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=85)
+@spaces.GPU(duration=65)
 def Piper(name,positive_prompt,negative,motion):
     global step
     global fps
@@ -188,12 +188,13 @@ repo = "ByteDance/AnimateDiff-Lightning"
 ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 base = "emilianJR/epiCRealism"
 #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
-    
-pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
+
+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).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)
-    
+
 mp.set_start_method("spawn", force=True)
 
 with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo: