SolitaryThinker
64cadaa0bf
- latent_shape(req, model): real Wan geometry (16ch; (T-1)//4+1, H/8, W/8) on the cuda backend; toy stand-in stays on accel/cpu. - stamp_wan21_checkpoints(card, model_root): map a root (local dir or HF id) onto the 3 components' ComponentSpec.checkpoint; build_wan21_card(checkpoint_root=...) optional.
10 lines
374 B
Python
10 lines
374 B
Python
"""Wan2.1-1.3B — bidirectional video diffusion (T2V)."""
|
|
from __future__ import annotations
|
|
|
|
from .card import build_wan21_card, stamp_wan21_checkpoints
|
|
from .loop import WanDenoiseLoop, latent_shape
|
|
from .program import build_wan_t2v_program
|
|
|
|
__all__ = ["build_wan21_card", "stamp_wan21_checkpoints", "build_wan_t2v_program",
|
|
"WanDenoiseLoop", "latent_shape"]
|