Skip to content

Commit c296b6f

Browse files
committed
fix naming
1 parent d993e16 commit c296b6f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/train_dreambooth_lora_hidream.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1649,7 +1649,7 @@ def get_sigmas(timesteps, n_dim=4, dtype=torch.float32):
16491649
hidden_states=noisy_model_input,
16501650
encoder_hidden_states=prompt_embeds,
16511651
pooled_embeds=pooled_prompt_embeds,
1652-
timestep=timesteps,
1652+
timesteps=timesteps,
16531653
img_sizes=img_sizes,
16541654
img_ids=img_ids,
16551655
return_dict=False,

0 commit comments

Comments
 (0)