Yaron Koresh commited on
Commit
b447299
·
verified ·
1 Parent(s): c350929

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -209,7 +209,7 @@ def main():
209
  adapter = MotionAdapter().to(device, dtype)
210
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device), strict=False)
211
 
212
- pipe = AnimateDiffPipeline.from_pretrained(base, unet=unet, motion_adapter=adapter, torch_dtype=dtype, variant="fp16").to(device, dtype)
213
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
214
 
215
  mp.set_start_method("spawn", force=True)
 
209
  adapter = MotionAdapter().to(device, dtype)
210
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device), strict=False)
211
 
212
+ pipe = AnimateDiffPipeline.from_pretrained(base, unet=unet, motion_adapter=adapter, torch_dtype=dtype, variant="fp16").to(dtype=dtype)
213
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
214
 
215
  mp.set_start_method("spawn", force=True)