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

Fix pylint error

Browse Source
This commit is contained in:
doctorpangloss 2025-01-16 14:09:37 -08:00
parent 005459ee77
commit a9347c6713
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/utils.py
View File

@ -57,9 +57,9 @@ if hasattr(torch.serialization, "add_safe_globals"): # TODO: this was added in
ModelCheckpoint.__module__ = "pytorch_lightning.callbacks.model_checkpoint" ModelCheckpoint.__module__ = "pytorch_lightning.callbacks.model_checkpoint"
from numpy.core.multiarray import scalar from numpy.core.multiarray import scalar # pylint: disable=no-name-in-module
from numpy import dtype from numpy import dtype
from numpy.dtypes import Float64DType from numpy.dtypes import Float64DType # pylint: disable=no-name-in-module
from _codecs import encode from _codecs import encode
torch.serialization.add_safe_globals([ModelCheckpoint, scalar, dtype, Float64DType, encode]) torch.serialization.add_safe_globals([ModelCheckpoint, scalar, dtype, Float64DType, encode])

3
requirements.txt
View File

@ -69,4 +69,5 @@ skia-python
pebble>=5.0.7 pebble>=5.0.7
openai openai
anthropic anthropic
humanize humanize
lightning