From c61c3fffad0349d3a1de41af162f2166f495192b Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 9 Oct 2024 09:43:50 -0700
Subject: [PATCH] Fix pylint import errors

---
 comfy_extras/nodes/nodes_torch_compile.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy_extras/nodes/nodes_torch_compile.py b/comfy_extras/nodes/nodes_torch_compile.py
index 23ffb8474..77f1f1294 100644
--- a/comfy_extras/nodes/nodes_torch_compile.py
+++ b/comfy_extras/nodes/nodes_torch_compile.py
@@ -85,14 +85,14 @@ class QuantizeModel(CustomNode):
             "final_layer.",
         }
         if strategy == "quanto":
-            from optimum.quanto import quantize, qint8
+            from optimum.quanto import quantize, qint8  # pylint: disable=import-error
             exclusion_list = [
                 name for name, module in unet.named_modules() if isinstance(module, LayerNorm) and module.weight is None
             ]
             quantize(unet, weights=qint8, activations=qint8, exclude=exclusion_list)
             _in_place_fixme = unet
         elif strategy == "torchao":
-            from torchao.quantization import quantize_, int8_dynamic_activation_int8_weight
+            from torchao.quantization import quantize_, int8_dynamic_activation_int8_weight  # pylint: disable=import-error
             model = model.clone()
             unet = model.get_model_object("diffusion_model")
             # todo: quantize quantizes in place, which is not desired