Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -22,13 +22,14 @@ from mmaudio.model.sequence_config import SequenceConfig
22
  from mmaudio.model.utils.features_utils import FeaturesUtils
23
  import tempfile
24
 
25
- torch.backends.cuda.matmul.allow_tf32 = True
26
- 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()
 
22
  from mmaudio.model.utils.features_utils import FeaturesUtils
23
  import tempfile
24
 
25
+ import torch
26
+ import logging
27
 
28
  log = logging.getLogger()
29
 
30
+ device = "cpu"
31
+ dtype = torch.float32
32
+
33
 
34
  model: ModelConfig = all_model_cfg['large_44k_v2']
35
  model.download_if_needed()