Setting steps to 16 causes crash

#17
by pravdomil - opened
src/ddim_v_hacked.py in ddim_sampling(self, cond, shape, x_T, ddim_use_original_steps, callback, timesteps, quantize_denoised, mask, x0, xtrg, noise_rescale, img_callback, log_every_t, temperature, noise_dropout, score_corrector, corrector_kwargs, unconditional_guidance_scale, unconditional_conditioning, dynamic_threshold, ucg_schedule, controller, strength)
    312                 # TODO: deterministic forward pass?
    313                 if type(mask) == list:
--> 314                     weight = mask[i]
    315                 else:
    316                     weight = mask

IndexError: list index out of range

Sign up or log in to comment