Skip to content

Commit e035a2c

Browse files
Fix config dpm (huggingface#3343)
1 parent e028b43 commit e035a2c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

schedulers/scheduling_dpmsolver_singlestep.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def set_timesteps(self, num_inference_steps: int, device: Union[str, torch.devic
241241
self.num_inference_steps = num_inference_steps
242242
# Clipping the minimum of all lambda(t) for numerical stability.
243243
# This is critical for cosine (squaredcos_cap_v2) noise schedule.
244-
clipped_idx = torch.searchsorted(torch.flip(self.lambda_t, [0]), self.lambda_min_clipped)
244+
clipped_idx = torch.searchsorted(torch.flip(self.lambda_t, [0]), self.config.lambda_min_clipped)
245245
timesteps = (
246246
np.linspace(0, self.config.num_train_timesteps - 1 - clipped_idx, num_inference_steps + 1)
247247
.round()[::-1][:-1]

0 commit comments

Comments
 (0)