oops
This commit is contained in:
parent
3395aa8ca0
commit
0b6812671c
3
nodes.py
3
nodes.py
@ -373,8 +373,9 @@ class MochiSampler:
|
|||||||
|
|
||||||
if opt_sigmas is not None:
|
if opt_sigmas is not None:
|
||||||
sigma_schedule = opt_sigmas.tolist()
|
sigma_schedule = opt_sigmas.tolist()
|
||||||
sigma_schedule.extend([1.0])
|
|
||||||
steps = len(sigma_schedule)
|
steps = len(sigma_schedule)
|
||||||
|
sigma_schedule.extend([1.0])
|
||||||
|
|
||||||
logging.info(f"Using sigma_schedule: {sigma_schedule}")
|
logging.info(f"Using sigma_schedule: {sigma_schedule}")
|
||||||
else:
|
else:
|
||||||
sigma_schedule = linear_quadratic_schedule(steps, 0.025)
|
sigma_schedule = linear_quadratic_schedule(steps, 0.025)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user