This commit is contained in:
kijai 2024-10-27 20:11:07 +02:00
parent 3395aa8ca0
commit 0b6812671c

View File

@ -373,8 +373,9 @@ class MochiSampler:
if opt_sigmas is not None:
sigma_schedule = opt_sigmas.tolist()
sigma_schedule.extend([1.0])
steps = len(sigma_schedule)
sigma_schedule.extend([1.0])
logging.info(f"Using sigma_schedule: {sigma_schedule}")
else:
sigma_schedule = linear_quadratic_schedule(steps, 0.025)