From 7e5587a5ac07be5621bfbbe24868b79e1cd0aede Mon Sep 17 00:00:00 2001 From: Will Berman Date: Fri, 25 Aug 2023 11:06:55 -0700 Subject: [PATCH] instance_prompt->class_prompt (#4784) --- examples/dreambooth/train_dreambooth_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth_lora.py b/examples/dreambooth/train_dreambooth_lora.py index ad862c697637..8b46d2143bd5 100644 --- 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