diff --git a/comfy/ldm/cosmos/position_embedding.py b/comfy/ldm/cosmos/position_embedding.py index 4d6a58dba..17ec96dda 100644 --- a/comfy/ldm/cosmos/position_embedding.py +++ b/comfy/ldm/cosmos/position_embedding.py @@ -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 diff --git a/comfy/model_base.py b/comfy/model_base.py index 23f0777f7..c0f240e2e 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -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 diff --git a/comfy/nodes/base_nodes.py b/comfy/nodes/base_nodes.py index 201655507..3f22f0778 100644 --- a/comfy/nodes/base_nodes.py +++ b/comfy/nodes/base_nodes.py @@ -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}")