From f4af3cf886caa4eceb4df84dbf8903cac2854d5e Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Thu, 18 Jun 2026 16:23:30 +0000 Subject: [PATCH] [feat] v2 registry: LTX-2/2.3 repo aliases -> by-id resolution 63/64 Adds explicit ModelEntry aliases for the LTX-2 (FastVideo/LTX2-Diffusers, LTX2-base, Lightricks/LTX-2 -> single-stage base) and LTX-2.3 (LTX2.3-Diffusers, LTX2.3-Distilled-Diffusers, LTX2.3-base, Lightricks/LTX-2.3, lightricks/ltx-2.3 -> the distilled joint-A/V card) naming variants of already-ported LTX checkpoints (the arch fallback also resolves LTX2Transformer3DModel from a root). v2 now resolves 63/64 fastvideo registry ids by exact id; the only remaining id, FastVideo/Wan2.1-VSA-T2V-14B-720P-Diffusers, is ENV-BLOCKED (VSA Sparse-Linear Attention kernels require nvcc, not built in this bring-up; it arch-resolves to the base Wan card but needs the VSA kernel build to run faithfully). v2 mini 240 passed, 2 skipped; pre-commit clean. --- v2/registry.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/v2/registry.py b/v2/registry.py index f0366540c..591707b5c 100644 --- a/v2/registry.py +++ b/v2/registry.py @@ -164,6 +164,14 @@ def _entries() -> list[ModelEntry]: ModelEntry(("Davids048/LTX2-Base-Diffusers", ), build_ltx2_base_card, build_ltx2_base_program), # LTX-2.3 distilled — single-stage, JOINT text->video+audio (separate connectors + gated attn): ModelEntry(("FastVideo/LTX-2.3-Distilled-Diffusers", ), build_ltx2_3_card, build_ltx2_3_program), + # LTX-2 / LTX-2.3 repo aliases (naming variants of the already-registered LTX checkpoints; the + # arch fallback also resolves LTX2Transformer3DModel from a checkpoint root, refining base-vs- + # two-stage by the spatial_upsampler). LTX-2 -> single-stage base; LTX-2.3 -> the distilled joint + # A/V card (a non-distilled 2.3 base checkpoint reuses it — override num_steps for many-step). + ModelEntry(("FastVideo/LTX2-Diffusers", "FastVideo/LTX2-base", "Lightricks/LTX-2"), build_ltx2_base_card, + build_ltx2_base_program), + ModelEntry(("FastVideo/LTX2.3-Diffusers", "FastVideo/LTX2.3-Distilled-Diffusers", "FastVideo/LTX2.3-base", + "Lightricks/LTX-2.3", "lightricks/ltx-2.3"), build_ltx2_3_card, build_ltx2_3_program), # Cosmos-Predict2-2B-Video2World — EDM-Karras denoiser (new CosmosDenoiseLoop + CosmosDiT adapter), # reusing the Wan VAE adapter + T5. Registered for t2v (video2world conditioning threads later). ModelEntry(("nvidia/Cosmos-Predict2-2B-Video2World", ), build_cosmos2_card, build_cosmos2_program),