Yaron Koresh commited on
Commit
cd81341
·
verified ·
1 Parent(s): c1d87d6

Update app.py

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