diff --git a/.gitignore b/.gitignore
index 82f9275..167139a 100644
--- a/.gitignore
+++ b/.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
diff --git a/mz_fluxext_core.py b/mz_fluxext_core.py
index 5021296..7ab171a 100644
--- a/mz_fluxext_core.py
+++ b/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")