Spaces:
Runtime error
Runtime error
Update compute.py
Browse files- compute.py +2 -2
compute.py
CHANGED
|
@@ -97,8 +97,8 @@ model = build_model(model_params, text_aligner, pitch_extractor, plbert)
|
|
| 97 |
_ = [model[key].eval() for key in model]
|
| 98 |
_ = [model[key].to(device) for key in model]
|
| 99 |
|
| 100 |
-
# params_whole = torch.load("
|
| 101 |
-
params_whole = torch.load(str(cached_path("hf://
|
| 102 |
params = params_whole['net']
|
| 103 |
|
| 104 |
for key in model:
|
|
|
|
| 97 |
_ = [model[key].eval() for key in model]
|
| 98 |
_ = [model[key].to(device) for key in model]
|
| 99 |
|
| 100 |
+
# params_whole = torch.load("epoch_2nd_00099.pth", map_location='cpu')
|
| 101 |
+
params_whole = torch.load(str(cached_path("hf://Bluebomber182/Judy-Hopps-StyleTTS2-Model/blob/main/epoch_2nd_00099.pth")), map_location='cpu')
|
| 102 |
params = params_whole['net']
|
| 103 |
|
| 104 |
for key in model:
|