Spaces:
Running
Running
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 |
|