Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
| 26 |
-
|
| 27 |
|
| 28 |
log = logging.getLogger()
|
| 29 |
|
| 30 |
-
device =
|
| 31 |
-
dtype = torch.
|
|
|
|
| 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()
|