diff --git a/int8_quant.py b/int8_quant.py
index 957609f..c512898 100644
--- a/int8_quant.py
+++ b/int8_quant.py
@@ -1035,6 +1035,9 @@ class INT8ModelPatcher(comfy.model_patcher.ModelPatcher):
                             module.weight_function = [f for f in getattr(module, "weight_function", []) if type(f).__name__ != "LowVramPatch"]
                         self.patch_weight_to_device(weight_key, device_to=device_to)
                     else:
+                        if hasattr(module, "weight_function"):
+                            module.weight_function = [f for f in getattr(module, "weight_function", []) if type(f).__name__ != "LowVramPatch"]
+                            
                         lowvram_patch = INT8LowVramPatch(
                             weight_key,
                             self.patches,