Hugging Face's logo

Spaces:
hkchengrex
/
MMAudio
Running on Zero

App Files Community
17
TheEliEngineer commited on
Commit
04cd4cd
·
verified ·
1 Parent(s): 6a6682f

Update app.py

Browse files
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()