From a13b9d3b548147c910b8855245be40fbeb772f8c Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Fri, 1 Nov 2024 11:13:08 -0700 Subject: [PATCH] Fix pylint error --- comfy_extras/nodes/nodes_torch_compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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])