From 70ab09732e7cfec0b19c497f823ddd1c8259dad0 Mon Sep 17 00:00:00 2001 From: "Vinh H. Pham" Date: Fri, 12 Jan 2024 13:30:55 +0700 Subject: [PATCH] Update examples/dreambooth/train_dreambooth.py Co-authored-by: Sayak Paul --- examples/dreambooth/train_dreambooth.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 532e134a61..d1e9c28e05 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -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]