instance_prompt->class_prompt (huggingface#4784)

jamarju · Aug 25, 2023 · 7e5587a · 7e5587a
1 parent dc8da1d
commit 7e5587a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py
@@ -999,7 +999,7 @@ def compute_text_embeddings(prompt):
             validation_prompt_encoder_hidden_states = None
 
         if args.class_prompt is not None:
-            pre_computed_class_prompt_encoder_hidden_states = compute_text_embeddings(args.instance_prompt)
+            pre_computed_class_prompt_encoder_hidden_states = compute_text_embeddings(args.class_prompt)
         else:
             pre_computed_class_prompt_encoder_hidden_states = None