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

Suppress with # pylint: disable=assignment-from-none because these are legit

Browse Source
This commit is contained in:
doctorpangloss 2024-06-19 20:47:35 -07:00
parent 185ba7e990
commit faba95d94a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_base.py
View File

@ -169,6 +169,7 @@ class BaseModel(torch.nn.Module):
data = torch.cat(cond_concat, dim=1)
out['c_concat'] = conds.CONDNoiseShape(data)
# pylint: disable=assignment-from-none
adm = self.encode_adm(**kwargs)
if adm is not None:
out['y'] = conds.CONDRegular(adm)
@ -522,6 +523,8 @@ class IP2P(BaseModel):
image = utils.resize_to_batch_size(image, noise.shape[0])
out['c_concat'] = conds.CONDNoiseShape(self.process_ip2p_image_in(image))
# pylint: disable=assignment-from-none
adm = self.encode_adm(**kwargs)
if adm is not None:
out['y'] = conds.CONDRegular(adm)