Skip to content

GitLab

"...en/git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "a2bc2e14b93b584bea6ba17bdaf7486c6ab808e2"
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
Hide 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