Skip to content

Commit d1b988b

Browse files
fix rf time scheduler problem (#14011)
fix rf-time scheduile problem Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
1 parent a1acc51 commit d1b988b

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)