From faba95d94a7e985606053ee9e02a835574f71be6 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Wed, 19 Jun 2024 20:47:35 -0700 Subject: [PATCH] Suppress with # pylint: disable=assignment-from-none because these are legit --- comfy/model_base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/comfy/model_base.py b/comfy/model_base.py index 7e7a8bae0..f9724dd7d 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -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)