LTX-2.5-diffusers-mini / ltx /probe_catalog.py
John6666's picture
Upload 39 files
e8b6587 verified
Raw
History Blame Contribute Delete
6.77 kB
"""Internal Maintainer/Research Probe profile catalog.
These are operator diagnostics, not fork-facing product configuration.
"""
from . import maintainer_probe as maintainer_probe_backend
from . import research_probe as research_probe_backend
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV = maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION = maintainer_probe_backend.FULL_SFT_30S_STAGE2_DIFFUSION_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE = maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_FRAMEWISE_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT = maintainer_probe_backend.FULL_SFT_30S_STAGE2_LATENT_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE = maintainer_probe_backend.FULL_SFT_30S_UPSAMPLE_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1 = maintainer_probe_backend.FULL_SFT_30S_FLF2V_STAGE1_PROFILE
MAINTAINER_PROBE_FULL_SFT_30S_STAGE1 = maintainer_probe_backend.FULL_SFT_30S_STAGE1_PROFILE
# P35 long-FLF2V structural gate: exercise the exact 721f Start+End condition-pipeline
# Stage 1 with two steps only. Long Stage 2/decode paths are already closed by P28/P33.
MAINTAINER_PROBE_CURRENT_GATES = (MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1,)
MAINTAINER_PROBE_CHOICES = [
("Current P35 gate · Long FLF2V · 30s Stage 1 structural scout", MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1),
("Closed P34 witness · Probe Progress smoke · 30s latent upsample only", MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE),
("Closed P33 witness · Full/SFT 30s · Stage 2 release → Diffusion Decoder", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION),
("Closed P28 witness · Full/SFT 30s · Stage 2 release → Conv VAE", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV),
("Fallback · Full/SFT 30s · Stage 2 release → framewise Conv VAE", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE),
("Diagnostic · Full/SFT 30s · Stage 2 latent-only", MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT),
("Closed witness rerun · Full/SFT 30s Stage 1 · 5-step", MAINTAINER_PROBE_FULL_SFT_30S_STAGE1),
]
def profile_description(profile_id) -> str:
selected = str(profile_id or "")
if selected == MAINTAINER_PROBE_FULL_SFT_30S_FLF2V_STAGE1:
return (
"**Current P35 gate · Long FLF2V 30s structural scout** \n"
"Full/SFT · Blue bird Start+End · 512×512 · 721f · seed 42 · no user LoRA. "
"Runs only **two Stage-1 denoising steps** through `LTX2ConditionPipeline` with Start index `0` and End index `-1`; "
"no latent upsample, Stage 2, decoder, MP4, or visual-quality claim. This isolates the only long-FLF2V-specific stage before a later product-quality run. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_FLF2V_STAGE1_ZEROGPU_DURATION_SECONDS}s**."
)
if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_DIFFUSION:
return f"**Closed P33 witness · 721f Stage 2 + Diffusion Decoder** \nFull/SFT · I2V · Cats · 512×512 · 721f · seed 42 · no user LoRA. P33 LIVE-pass済みの再現用profileです。Two Stage-1 steps create the exact 721f latent shape; real Stage 2 + Diffusion Decoder runs, with no 30-step Stage 1 or MP4 encode. Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_DIFFUSION_ZEROGPU_DURATION_SECONDS}s**."
if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV:
return (
"**Closed P28 witness rerun · 721f Stage 2 + Conv VAE measurement** \n"
"Fixed contract: **Full/SFT · I2V · Cats · 512×512 · 721f · seed 42 · no user LoRA**. "
"Two Stage-1 steps create the exact temporal latent shape, then the real ×2 latent upsampler and internal distilled-LoRA Stage 2 run. Stage 2 returns latents, the internal adapter is released, earlier latent references are dropped, and only then the normal Conv VAE/audio decode runs. P28R4 already LIVE-passed this profile; it is retained for reproducibility. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_ZEROGPU_DURATION_SECONDS}s**. It does not repeat the expensive 30-step Stage 1 and does not MP4-encode."
)
if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_CONV_FRAMEWISE:
return (
"**Fallback diagnostic · post-Stage2 release + framewise Conv VAE** \n"
"Same 721f two-step preparation and real Stage 2, but the Conv VAE temporarily enables pinned LTX2 temporal framewise decoding after Stage-2 adapter release. This is not the current product gate; use it only if the product-equivalent release-before-decode gate still OOMs. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_CONV_FRAMEWISE_ZEROGPU_DURATION_SECONDS}s**."
)
if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE2_LATENT:
return (
"**Optional diagnostic · 721f Stage 2 latent-only** \n"
"Same two-step Stage-1 + ×2 upsample preparation, then the real internal distilled-LoRA Stage 2 returns latents instead of decoding. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE2_LATENT_ZEROGPU_DURATION_SECONDS}s**. Use this only if the Conv-VAE gate needs Stage-2/decode isolation."
)
if selected == MAINTAINER_PROBE_FULL_SFT_30S_UPSAMPLE:
return (
"**Closed P34 witness · Probe Progress visibility smoke** \n"
"Reuses the closed 721f two-step latent-upsample witness only. Stage 1 now emits explicit step callbacks and the event pins Gradio Progress to this status surface. "
"Confirm that progress visibly advances while the Probe runs; no Stage 2, decoder, MP4, or visual-quality judgment is required. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_UPSAMPLE_ZEROGPU_DURATION_SECONDS}s**."
)
if selected == MAINTAINER_PROBE_FULL_SFT_30S_STAGE1:
return (
"**Closed witness rerun · Stage-1 timing scout** \n"
"Full/SFT · I2V · Cats · 512×512 · 721f · 5 Stage-1 steps · seed 42 · no user LoRA · latent output only. "
f"Reservation: **{maintainer_probe_backend.FULL_SFT_30S_STAGE1_ZEROGPU_DURATION_SECONDS}s**. This profile is retained for reproducibility; it is not the current gate."
)
return "Select a Probe profile. No Probe runs until **Run Probe** is pressed."
RESEARCH_PROBE_CHOICES = [
("Research · 30s Stage 1 · 8-step timing curve", research_probe_backend.FULL_SFT_30S_STAGE1_8STEP_PROFILE),
("Research · 30s Stage 1 · 12-step timing curve", research_probe_backend.FULL_SFT_30S_STAGE1_12STEP_PROFILE),
("Research · 25s Stage 1 · 5-step timing curve", research_probe_backend.FULL_SFT_25S_STAGE1_5STEP_PROFILE),
]