kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[LCM] Better error message (#5788)

Browse Source
This commit is contained in:
Patrick von Platen
2023-11-14 12:08:15 +01:00
committed by GitHub
parent 81df9c85de
commit 0a0ebc7cb4
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/schedulers/scheduling_lcm.py
+6
View File
@@ -378,6 +378,12 @@ class LCMScheduler(SchedulerMixin, ConfigMixin):
# LCM Training Steps Schedule # LCM Training Steps Schedule
lcm_origin_timesteps = np.asarray(list(range(1, int(original_steps * strength) + 1))) * c - 1 lcm_origin_timesteps = np.asarray(list(range(1, int(original_steps * strength) + 1))) * c - 1
skipping_step = len(lcm_origin_timesteps) // num_inference_steps skipping_step = len(lcm_origin_timesteps) // num_inference_steps
if skipping_step < 1:
raise ValueError(
f"The combination of `original_steps x strength`: {original_steps} x {strength} is smaller than `num_inference_steps`: {num_inference_steps}. Make sure to either reduce `num_inference_steps` to a value smaller than {int(original_steps * strength)} or increase `strength` to a value higher than {float(num_inference_steps / original_steps)}."
)
# LCM Inference Steps Schedule # LCM Inference Steps Schedule
timesteps = lcm_origin_timesteps[::-skipping_step][:num_inference_steps] timesteps = lcm_origin_timesteps[::-skipping_step][:num_inference_steps]