From 30e47f46902bab55f1c840f124f214f6198de107 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Mon, 14 Oct 2024 09:48:16 -0700 Subject: [PATCH] Fix pylint errors --- comfy_extras/nodes/nodes_torch_compile.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/comfy_extras/nodes/nodes_torch_compile.py b/comfy_extras/nodes/nodes_torch_compile.py index 425d10cfd..fce477be0 100644 --- a/comfy_extras/nodes/nodes_torch_compile.py +++ b/comfy_extras/nodes/nodes_torch_compile.py @@ -110,7 +110,7 @@ class QuantizeModel(CustomNode): quantize(unet, weights=qint8, activations=qint8, exclude=exclusion_list) _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.quantization import quantize_, int8_dynamic_activation_int8_weight, unwrap_tensor_subclass, autoquant # pylint: disable=import-error model = model.clone() self.warn_in_place(model) unet = model.get_model_object("diffusion_model") @@ -123,7 +123,6 @@ class QuantizeModel(CustomNode): else: quantize_(unet, int8_dynamic_activation_int8_weight(), device=model_management.get_torch_device(), set_inductor_config=False) _in_place_fixme = unet - from torchao.utils import unwrap_tensor_subclass unwrap_tensor_subclass(_in_place_fixme) else: raise ValueError(f"unknown strategy {strategy}")