mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
add workaround for 'Some tensors share memory' error #1614
This commit is contained in:
@@ -412,6 +412,10 @@ def main(args):
|
||||
state_dict = convert_ai_toolkit_to_sd_scripts(state_dict)
|
||||
elif args.src == "sd-scripts" and args.dst == "ai-toolkit":
|
||||
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:
|
||||
raise NotImplementedError(f"Conversion from {args.src} to {args.dst} is not supported")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user