instance_prompt->class_prompt (#4784)

7e5587a5 · Will Berman · GitHub · dc8da1d4 · 7e5587a5
Unverified Commit 7e5587a5 authored Aug 25, 2023 by Will Berman Committed by GitHub Aug 25, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/dreambooth/train_dreambooth_lora.py examples/dreambooth/train_dreambooth_lora.py +1 -1

No files found.
--- a/examples/dreambooth/train_dreambooth_lora.py
+++ b/examples/dreambooth/train_dreambooth_lora.py
@@ -999,7 +999,7 @@ def main(args):
            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