styletts2_Japanese

Runtime error

Respair commited on Mar 28

Commit

5e89881

•

1 Parent(s): 8e33a4b

Update styletts2importable.py

Files changed (1) hide show

styletts2importable.py CHANGED Viewed

@@ -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=1.0, rho=9.0), # empirical parameters
     clamp=False
 )

 sampler = DiffusionSampler(
     model.diffusion.diffusion,
     sampler=ADPM2Sampler(),
+    sigma_schedule=KarrasSchedule(sigma_min=0.0001, sigma_max=0.57, rho=9.), # empirical parameters
     clamp=False
 )