Fix pylint error

This commit is contained in:
doctorpangloss 2024-11-01 11:13:08 -07:00
parent 31eacb6ac9
commit a13b9d3b54

View File

@ -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])