wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix pylint error

Browse Source
This commit is contained in:
doctorpangloss 2024-11-01 11:13:08 -07:00
parent 31eacb6ac9
commit a13b9d3b54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes/nodes_torch_compile.py
View File

@ -176,7 +176,7 @@ class QuantizeModel(CustomNode):
_in_place_fixme = unet _in_place_fixme = unet
elif "torchao" in strategy: 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, 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) self.warn_in_place(model)
model_management.load_models_gpu([model]) model_management.load_models_gpu([model])