'update'

.gitignore

@@ -160,3 +160,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+copytoww.bat
+exclude.txt

mz_fluxext_core.py

@@ -9,10 +9,12 @@ import safetensors
 
 
 from torch import Tensor, nn
+import copy
 
 
 def Flux1PartialLoad_Patch(args={}):
     model = args.get("model")
+
     double_blocks_cuda_size = args.get("double_blocks_cuda_size")
     single_blocks_cuda_size = args.get("single_blocks_cuda_size")