Spaces:
Running
Running
Update styletts2importable.py
Browse files- 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=
|
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 |
|