diff --git a/comfy_extras/nodes/nodes_torch_compile.py b/comfy_extras/nodes/nodes_torch_compile.py
index d158c058c..fcb37459d 100644
--- a/comfy_extras/nodes/nodes_torch_compile.py
+++ b/comfy_extras/nodes/nodes_torch_compile.py
@@ -176,7 +176,7 @@ class QuantizeModel(CustomNode):
             _in_place_fixme = unet
         elif "torchao" in strategy:
             from torchao.quantization import quantize_, int8_dynamic_activation_int8_weight, autoquant  # pylint: disable=import-error
-            from torchao.utils import unwrap_tensor_subclass
+            from torchao.utils import unwrap_tensor_subclass  # pylint: disable=import-error
             self.warn_in_place(model)
             model_management.load_models_gpu([model])