Respair commited on
Commit
d919442
1 Parent(s): f4e7b50

Update styletts2importable.py

Browse files
Files changed (1) hide show
  1. styletts2importable.py +1 -1
styletts2importable.py CHANGED
@@ -761,7 +761,7 @@ from Modules.diffusion.sampler import DiffusionSampler, ADPM2Sampler, KarrasSche
761
  sampler = DiffusionSampler(
762
  model.diffusion.diffusion,
763
  sampler=ADPM2Sampler(),
764
- sigma_schedule=KarrasSchedule(sigma_min=0.0001, sigma_max=0.68, rho=4.6), # empirical parameters
765
  clamp=False
766
  )
767
 
 
761
  sampler = DiffusionSampler(
762
  model.diffusion.diffusion,
763
  sampler=ADPM2Sampler(),
764
+ sigma_schedule=KarrasSchedule(sigma_min=0.0001, sigma_max=3., rho=9.), # empirical parameters
765
  clamp=False
766
  )
767