mirror of
https://github.com/wassname/peft.git
synced 2026-10-29 13:11:48 +08:00
fix ci dreambooth
This commit is contained in:
1 parent
098962fa65
commit
891584c8d9
1 file changed
+3
-1
@@ -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"))
|
||||
|
||||
Reference in new issue
Block a user