Hugging Face's logo

Spaces:
hkchengrex
/
MMAudio
Running on Zero

App Files Community
17

Update app.py

#17
by TheEliEngineer - opened
base: refs/heads/main
←
from: refs/pr/17
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ torch.backends.cudnn.allow_tf32 = True
27
 
28
  log = logging.getLogger()
29
 
30
- device = 'cuda'
31
- dtype = torch.bfloat16
32
 
33
  model: ModelConfig = all_model_cfg['large_44k_v2']
34
  model.download_if_needed()
 
27
 
28
  log = logging.getLogger()
29
 
30
+ device = 'cpu'
31
+ dtype = torch.float32 # safer on CPU
32
 
33
  model: ModelConfig = all_model_cfg['large_44k_v2']
34
  model.download_if_needed()