wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-02 17:50:19 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Fix pylint errors (they were real, as they usually are)

Browse Source
This commit is contained in:
doctorpangloss 2025-01-28 17:16:15 -08:00
parent d24098cd9b
commit 95a12f42e2
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/cosmos/position_embedding.py
View File

@ -50,7 +50,7 @@ class VideoPositionEmb(nn.Module):
return embeddings
def generate_embeddings(self, B_T_H_W_C: torch.Size, fps=Optional[torch.Tensor], device=None):
def generate_embeddings(self, B_T_H_W_C: torch.Size, fps=Optional[torch.Tensor], device=None, dtype=None, **kwargs):
raise NotImplementedError

2
comfy/model_base.py
View File

@ -580,7 +580,7 @@ class SD_X4Upscaler(BaseModel):
cross_attn = kwargs.get("cross_attn", None)
if cross_attn is not None:
out['c_crossattn'] = comfy.conds.CONDCrossAttn(cross_attn)
out['c_crossattn'] = conds.CONDCrossAttn(cross_attn)
return out

2
comfy/nodes/base_nodes.py
View File

@ -973,7 +973,7 @@ class CLIPLoader:
elif type == "pixart":
clip_type = sd.CLIPType.PIXART
elif type == "cosmos":
clip_type = comfy.sd.CLIPType.COSMOS
clip_type = sd.CLIPType.COSMOS
else:
logging.warning(f"Unknown clip type argument passed: {type} for model {clip_name}")