Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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).to(device
|
| 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(device)
|
| 219 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
|
| 220 |
|
| 221 |
mp.set_start_method("spawn", force=True)
|