Skip to content

GitLab

Unverified Commit a68f5062 authored by Liang Hou's avatar Liang Hou Committed by GitHub
Browse files

style(sdxl): remove identity assignments (#5418)

parent b864d674
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/text_to_image/train_text_to_image_sdxl.py
View file @ a68f5062
...@@ -1038,7 +1038,6 @@ def main(args): ...@@ -1038,7 +1038,6 @@ def main(args):
prompt_embeds = batch["prompt_embeds"].to(accelerator.device) prompt_embeds = batch["prompt_embeds"].to(accelerator.device)
pooled_prompt_embeds = batch["pooled_prompt_embeds"].to(accelerator.device) pooled_prompt_embeds = batch["pooled_prompt_embeds"].to(accelerator.device)
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