wassname/peft
1
0 Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-10-30 13:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #224 from younesbelkada/fix-ci-dreambooth

Browse Source 
Fix failing test on `main`
This commit is contained in:
Sourab Mangrulkar authored and GitHub committed 2023-03-29 17:15:42 +05:30
2 parents 098962fa65 891584c8d9
commit 51f49a5fe4
1 file changed
+3 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/lora_dreambooth/train_dreambooth.py
+3 -1
View File
@@ -1063,7 +1063,9 @@ def main(args):
)
text_encoder_state_dict = {f"text_encoder_{k}": v for k, v in text_encoder_state_dict.items()}
state_dict.update(text_encoder_state_dict)
lora_config["text_encoder_peft_config"] = unwarpped_text_encoder.get_peft_config_as_dict(inference=True)
lora_config["text_encoder_peft_config"] = unwarpped_text_encoder.get_peft_config_as_dict(
inference=True
)
accelerator.print(state_dict)
accelerator.save(state_dict, os.path.join(args.output_dir, f"{args.instance_prompt}_lora.pt"))