Spaces:
Sleeping
Sleeping
GaboChoropan
commited on
Update _run.py
Browse files
_run.py
CHANGED
@@ -99,7 +99,7 @@ model = build_model(model_params, text_aligner, pitch_extractor, plbert)
|
|
99 |
_ = [model[key].eval() for key in model]
|
100 |
_ = [model[key].to(device) for key in model]
|
101 |
|
102 |
-
params_whole = torch.load("
|
103 |
params = params_whole['net']
|
104 |
|
105 |
for key in model:
|
|
|
99 |
_ = [model[key].eval() for key in model]
|
100 |
_ = [model[key].to(device) for key in model]
|
101 |
|
102 |
+
params_whole = torch.load("epoch_2nd_00099.pth", map_location='cpu')
|
103 |
params = params_whole['net']
|
104 |
|
105 |
for key in model:
|