Closed Chronyaa closed 1 year ago
这里仍然保存了转换前的sd_TE而不是转换后的state
sd_TE
state
diff --git a/hcpdiff/tools/lora_convert.py b/hcpdiff/tools/lora_convert.py index 8242ea9..25228eb 100644 --- a/hcpdiff/tools/lora_convert.py +++ b/hcpdiff/tools/lora_convert.py @@ -83,5 +83,5 @@ if __name__ == '__main__': sd_unet = ckpt_manager.load_ckpt(args.lora_path) sd_TE = ckpt_manager.load_ckpt(args.lora_path_TE) state = converter.convert_to_webui(sd_unet['lora'], sd_TE['lora']) - ckpt_manager._save_ckpt(sd_TE, save_path=args.dump_path) + ckpt_manager._save_ckpt(state, save_path=args.dump_path) print('save lora to:', args.dump_path) \ No newline at end of file
感谢反馈bug,已经修复
这里仍然保存了转换前的
sd_TE
而不是转换后的state