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