Skip to content

GitLab

Unverified Commit 7e5587a5 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

instance_prompt->class_prompt (#4784)

parent dc8da1d4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/dreambooth/train_dreambooth_lora.py
View file @ 7e5587a5
...@@ -999,7 +999,7 @@ def main(args): ...@@ -999,7 +999,7 @@ def main(args):
validation_prompt_encoder_hidden_states = None validation_prompt_encoder_hidden_states = None
if args.class_prompt is not 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: else:
pre_computed_class_prompt_encoder_hidden_states = None pre_computed_class_prompt_encoder_hidden_states = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment