Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ model.cfm.estimator.setup_caches(max_batch_size=1, max_seq_length=8192)
|
|
31 |
from modules.campplus.DTDNN import CAMPPlus
|
32 |
|
33 |
campplus_model = CAMPPlus(feat_dim=80, embedding_size=192)
|
34 |
-
campplus_model.load_state_dict(torch.load(config['model_params']['style_encoder']['campplus_path']
|
35 |
campplus_model.eval()
|
36 |
campplus_model.to(device)
|
37 |
|
|
|
31 |
from modules.campplus.DTDNN import CAMPPlus
|
32 |
|
33 |
campplus_model = CAMPPlus(feat_dim=80, embedding_size=192)
|
34 |
+
campplus_model.load_state_dict(torch.load(config['model_params']['style_encoder']['campplus_path'], map_location='cpu'))
|
35 |
campplus_model.eval()
|
36 |
campplus_model.to(device)
|
37 |
|