Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,8 +23,8 @@ motion_loaded = None
|
|
| 23 |
if not torch.cuda.is_available():
|
| 24 |
raise NotImplementedError("No GPU detected!")
|
| 25 |
|
| 26 |
-
device = "
|
| 27 |
-
dtype =
|
| 28 |
pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
|
| 29 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
|
| 30 |
|
|
|
|
| 23 |
if not torch.cuda.is_available():
|
| 24 |
raise NotImplementedError("No GPU detected!")
|
| 25 |
|
| 26 |
+
device = "cpu"
|
| 27 |
+
dtype = None
|
| 28 |
pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
|
| 29 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
|
| 30 |
|