VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add workaround for 'Some tensors share memory' error #1614

Browse Source
This commit is contained in:
Kohya S
2024-09-28 20:57:27 +09:00
parent a9aa52658a
commit 822fe57859
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
networks/convert_flux_lora.py
View File

@@ -412,6 +412,10 @@ def main(args):
state_dict = convert_ai_toolkit_to_sd_scripts(state_dict) state_dict = convert_ai_toolkit_to_sd_scripts(state_dict)
elif args.src == "sd-scripts" and args.dst == "ai-toolkit": elif args.src == "sd-scripts" and args.dst == "ai-toolkit":
state_dict = convert_sd_scripts_to_ai_toolkit(state_dict) state_dict = convert_sd_scripts_to_ai_toolkit(state_dict)
# eliminate 'shared tensors'
for k in list(state_dict.keys()):
state_dict[k] = state_dict[k].detach().clone()
else: else:
raise NotImplementedError(f"Conversion from {args.src} to {args.dst} is not supported") raise NotImplementedError(f"Conversion from {args.src} to {args.dst} is not supported")