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

Fix pylint import errors

Browse Source
This commit is contained in:
doctorpangloss 2024-10-09 09:43:50 -07:00
parent 99f0fa8b50
commit c61c3fffad
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes/nodes_torch_compile.py
View File

@ -85,14 +85,14 @@ class QuantizeModel(CustomNode):
"final_layer.", "final_layer.",
} }
if strategy == "quanto": if strategy == "quanto":
from optimum.quanto import quantize, qint8 from optimum.quanto import quantize, qint8 # pylint: disable=import-error
exclusion_list = [ exclusion_list = [
name for name, module in unet.named_modules() if isinstance(module, LayerNorm) and module.weight is None 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) quantize(unet, weights=qint8, activations=qint8, exclude=exclusion_list)
_in_place_fixme = unet _in_place_fixme = unet
elif strategy == "torchao": 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() model = model.clone()
unet = model.get_model_object("diffusion_model") unet = model.get_model_object("diffusion_model")
# todo: quantize quantizes in place, which is not desired # todo: quantize quantizes in place, which is not desired