AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update examples/dreambooth/train_dreambooth.py

Browse Source 
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
Vinh H. Pham
2024-01-12 13:30:55 +07:00
committed by GitHub
parent 362461ba04
commit 70ab09732e
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/dreambooth/train_dreambooth.py
View File

@@ -789,10 +789,7 @@ def encode_prompt(text_encoder, input_ids, attention_mask, text_encoder_use_atte
else:
attention_mask = None
prompt_embeds = text_encoder(
text_input_ids,
attention_mask=attention_mask,
return_dict=False,
prompt_embeds = text_encoder(text_input_ids, attention_mask=attention_mask, return_dict=False)
)
prompt_embeds = prompt_embeds[0]