Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "ba4348d9a791c63b2f7e7cdc45df4629956eb8b3"
Unverified Commit 20e426cb authored by Chengrui Wang's avatar Chengrui Wang Committed by GitHub
Browse files

Fix bug in train_dreambooth_lora (#3183) 

* Update train_dreambooth_lora.py

fix bug

* Update train_dreambooth_lora.py
parent 90eac14f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/dreambooth/train_dreambooth_lora.py
View file @ 20e426cb
...@@ -743,7 +743,7 @@ def main(args): ...@@ -743,7 +743,7 @@ def main(args):
) )
temp_pipeline._modify_text_encoder(text_lora_attn_procs) temp_pipeline._modify_text_encoder(text_lora_attn_procs)
text_encoder = temp_pipeline.text_encoder text_encoder = temp_pipeline.text_encoder
accelerator.register_for_checkpointing(unet_lora_layers) accelerator.register_for_checkpointing(text_encoder_lora_layers)
del temp_pipeline del temp_pipeline
if args.scale_lr: if args.scale_lr:
......
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