K00B404 commited on
Commit
20ff640
1 Parent(s): b8bee49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "cuda"
27
- dtype = torch.float16
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