Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
Dango233 authored Nov 7, 2024
1 parent e54462a commit bafd10d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion library/sd3_train_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -980,7 +980,7 @@ def get_noisy_model_input_and_timesteps(
indices = (u * (t_max-t_min) + t_min).long()
timesteps = indices.to(device=device, dtype=dtype)

# sigmas according to dlowmatching
# sigmas according to flowmatching
sigmas = timesteps / 1000
sigmas = sigmas.view(-1,1,1,1)
noisy_model_input = sigmas * noise + (1.0 - sigmas) * latents
Expand Down

0 comments on commit bafd10d

Please sign in to comment.