Skip to content

Commit 5a831a3

Browse files
authored
Merge branch 'main' into add-joyimage-edit-plus
2 parents 967a4ba + d1b988b commit 5a831a3

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

src/diffusers/schedulers/scheduling_flow_match_euler_discrete.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -364,10 +364,7 @@ def set_timesteps(
364364

365365
# 5. Convert sigmas and timesteps to tensors and move to specified device
366366
sigmas = torch.from_numpy(sigmas).to(dtype=torch.float32, device=device)
367-
if not is_timesteps_provided:
368-
timesteps = sigmas * self.config.num_train_timesteps
369-
else:
370-
timesteps = torch.from_numpy(timesteps).to(dtype=torch.float32, device=device)
367+
timesteps = sigmas * self.config.num_train_timesteps
371368

372369
# 6. Append the terminal sigma value.
373370
# If a model requires inverted sigma schedule for denoising but timesteps without inversion, the

0 commit comments

Comments
 (0)