Update styletts2importable.py

styletts2importable.py

@@ -735,7 +735,7 @@ _ = [model[key].eval() for key in model]
 _ = [model[key].to(device) for key in model]
 
 # params_whole = torch.load("Models/LibriTTS/epochs_2nd_00020.pth", map_location='cpu')
-params_whole = torch.load("Models/Kaede.pth", map_location='cpu')
+params_whole = torch.load("Models/epoch_2nd_00046_NO_SLM.pth", map_location='cpu')
 params = params_whole['net']
 
 for key in model:
@@ -761,7 +761,7 @@ from Modules.diffusion.sampler import DiffusionSampler, ADPM2Sampler, KarrasSche
 sampler = DiffusionSampler(
     model.diffusion.diffusion,
     sampler=ADPM2Sampler(),
-    sigma_schedule=KarrasSchedule(sigma_min=0.0001, sigma_max=0.57, rho=9.), # empirical parameters
+    sigma_schedule=KarrasSchedule(sigma_min=0.0001, sigma_max=0.68, rho=4.6), # empirical parameters
     clamp=False
 )