wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-14 23:42:35 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix pylint errors

Browse Source
This commit is contained in:
doctorpangloss 2024-10-14 09:48:16 -07:00
parent c0d1c9f96d
commit 30e47f4690
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes/nodes_torch_compile.py
View File

@ -110,7 +110,7 @@ class QuantizeModel(CustomNode):
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 "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, unwrap_tensor_subclass, autoquant # pylint: disable=import-error
model = model.clone() model = model.clone()
self.warn_in_place(model) self.warn_in_place(model)
unet = model.get_model_object("diffusion_model") unet = model.get_model_object("diffusion_model")
@ -123,7 +123,6 @@ class QuantizeModel(CustomNode):
else: else:
quantize_(unet, int8_dynamic_activation_int8_weight(), device=model_management.get_torch_device(), set_inductor_config=False) quantize_(unet, int8_dynamic_activation_int8_weight(), device=model_management.get_torch_device(), set_inductor_config=False)
_in_place_fixme = unet _in_place_fixme = unet
from torchao.utils import unwrap_tensor_subclass
unwrap_tensor_subclass(_in_place_fixme) unwrap_tensor_subclass(_in_place_fixme)
else: else:
raise ValueError(f"unknown strategy {strategy}") raise ValueError(f"unknown strategy {strategy}")