SolitaryThinker
4e331eb7ff
Mechanical conversion of every relative import under v2/ to an absolute v2.* path (from .x / ..x / ...x -> from v2.<pkg>.x) so imports are unambiguous, grep-able, and stable when code is copied/moved between entrypoints (VideoGenerator, CLI, server). Surgical prefix-only rewrite: only the 'from <dots><module>' prefix changed — import names, parentheses, multi-line formatting, comments, and ordering are byte-for-byte preserved (no collapsing, no reorder, no unrelated reformatting). - 471 imports across 125 files; v2/tests/ was already absolute (untouched). - Validated: all 125 files compile, every 'from v2.* import' target resolves to a real module/package, zero relative imports remain (full sweep), CPU mini suite green (202 passed, 2 skipped).
24 lines
1.1 KiB
Python
24 lines
1.1 KiB
Python
"""Wan-causal streaming program: text_encode → chunk_rollout → vae_decode (design_v3 §4, §15e)."""
|
|
from __future__ import annotations
|
|
|
|
from v2.program import ComponentNode, ModelLoopNode, Program, ProgramKind
|
|
from v2.models.common import text_encode_node_fn as _text_encode
|
|
|
|
|
|
def _vae_decode(instance, slots, request, ctx) -> None:
|
|
latents = slots["rollout_out"]["latents"]
|
|
slots["video"] = instance.component("vae").decode(latents) if latents is not None else None
|
|
|
|
|
|
def build_wan_causal_program() -> Program:
|
|
return Program(
|
|
program_id="wan_causal.stream", kind=ProgramKind.INLINE,
|
|
nodes=[
|
|
ComponentNode("text_encode", fn=_text_encode, writes=("text_embeds", "neg_text_embeds")),
|
|
ModelLoopNode("rollout", loop_id="chunk_rollout", output_slot="rollout_out",
|
|
reads=("text_embeds",), writes=("rollout_out",)),
|
|
ComponentNode("vae_decode", fn=_vae_decode, reads=("rollout_out",), writes=("video",)),
|
|
],
|
|
output_artifacts={"video": "video", "latents": "rollout_out"},
|
|
).validate()
|