Spaces:
Sleeping
Sleeping
GaboChoropan
commited on
Commit
•
82dd451
1
Parent(s):
3391343
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:
|