Skip to content

Commit a65d006

Browse files
committed
Fix #13394: DDPMScheduler set_timesteps 缺少 num_inference_steps <= 0 验证
1 parent 7bf0000 commit a65d006

2 files changed

Lines changed: 14 additions & 0 deletions

File tree

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
timesteps = np.array(timesteps, dtype=np.int64)
2+
self.custom_timesteps = True
3+
else:
4+
if num_inference_steps <= 0:
5+
raise ValueError(
6+
f"`num_inference_steps` must be a positive integer, but got {num_inference_steps}."
7+
)
8+
if num_inference_steps > self.config.num_train_timesteps:
9+
raise ValueError(
10+
f"`num_inference_steps`: {num_inference_steps} cannot be larger than `self.config.train_timesteps`:"

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -308,6 +308,10 @@ def set_timesteps(
308308
timesteps = np.array(timesteps, dtype=np.int64)
309309
self.custom_timesteps = True
310310
else:
311+
if num_inference_steps <= 0:
312+
raise ValueError(
313+
f"`num_inference_steps` must be a positive integer, but got {num_inference_steps}."
314+
)
311315
if num_inference_steps > self.config.num_train_timesteps:
312316
raise ValueError(
313317
f"`num_inference_steps`: {num_inference_steps} cannot be larger than `self.config.train_timesteps`:"

0 commit comments

Comments
 (0)