wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-12 01:07:30 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Update model_base.py

Browse Source
This commit is contained in:
kijai 2026-02-24 17:29:34 +02:00
parent 41eccdd8af
commit 15ddf5f6e5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_base.py
View File

@ -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):