GaboChoropan commited on
Commit
82dd451
1 Parent(s): 3391343

Update compute.py

Browse files
Files changed (1) hide show
  1. 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("Models/LibriTTS/epochs_2nd_00020.pth", map_location='cpu')
101
- params_whole = torch.load(str(cached_path("hf://yl4579/StyleTTS2-LibriTTS/Models/LibriTTS/epochs_2nd_00020.pth")), map_location='cpu')
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: