lora_convert.py --to_webui没有保存转换后的state而是保存了转换前的TextEncoder的state

[Chronyaa]

这里仍然保存了转换前的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

[IrisRainbowNeko]

感谢反馈bug，已经修复