diff --git a/comfy/model_base.py b/comfy/model_base.py index 5b844340e..4821dbb12 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -1513,6 +1513,10 @@ class WAN21_SCAIL(WAN21): if ref_latents is not None: out['reference_latent'] = list([1, 16, sum(map(lambda a: math.prod(a.size()), ref_latents)) // 16]) + pose_latents = kwargs.get("pose_video_latent", None) + if pose_latents is not None: + out['pose_latents'] = pose_latents.shape + return out class Hunyuan3Dv2(BaseModel):