Yaron Koresh commited on
Commit
293f346
·
verified ·
1 Parent(s): 6b11cde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -182,7 +182,7 @@ def main():
182
  adapter = MotionAdapter().to(device, dtype)
183
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
184
 
185
- pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, unet=unet, torch_dtype=dtype, variant="fp16", device_map=device)
186
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
187
 
188
  mp.set_start_method("spawn", force=True)
 
182
  adapter = MotionAdapter().to(device, dtype)
183
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
184
 
185
+ pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, unet=unet, torch_dtype=dtype, variant="fp16")
186
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
187
 
188
  mp.set_start_method("spawn", force=True)