Skip to content

GitLab

Unverified Commit 078df46b authored by Wang Qiang's avatar Wang Qiang Committed by GitHub
Browse files

An invalid clerical error in sdxl finetune (#4608)

parent 15782fd5
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/text_to_image/train_text_to_image_lora_sdxl.py
View file @ 078df46b
...@@ -1051,7 +1051,6 @@ def main(args): ...@@ -1051,7 +1051,6 @@ def main(args):
text_input_ids_list=[batch["input_ids_one"], batch["input_ids_two"]], text_input_ids_list=[batch["input_ids_one"], batch["input_ids_two"]],
) )
unet_added_conditions.update({"text_embeds": pooled_prompt_embeds}) unet_added_conditions.update({"text_embeds": pooled_prompt_embeds})
prompt_embeds = prompt_embeds
model_pred = unet( model_pred = unet(
noisy_model_input, timesteps, prompt_embeds, added_cond_kwargs=unet_added_conditions noisy_model_input, timesteps, prompt_embeds, added_cond_kwargs=unet_added_conditions
).sample ).sample
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment