add workaround for 'Some tensors share memory' error kohya-ss#1614

sdbds · Sep 28, 2024 · 822fe57 · 822fe57
1 parent a9aa526
commit 822fe57
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/networks/convert_flux_lora.py b/networks/convert_flux_lora.py
@@ -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")