Fix return statement

This commit is contained in:
Yousef R. Gamaleldin 2026-04-12 20:21:19 +02:00 committed by GitHub
parent e2171627c3
commit f4ae7b8391
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1589,6 +1589,7 @@ class WAN21_SCAIL(WAN21):
pose_latents = kwargs.get("pose_video_latent", None)
if pose_latents is not None:
out['pose_latents'] = [pose_latents.shape[0], 20, *pose_latents.shape[2:]]
return out
class Hunyuan3Dv2(BaseModel):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):