[new-model] Add Cosmos Predict2.5 2B distilled T2W inference - #1767

Open

Mister-Raggs wants to merge 16 commits into

hao-ai-lab:mainfrom

Mister-Raggs:feat/cosmos25-distilled-sampler

Commits on Aug 27, 2026

[feat] add Cosmos 2.5 distilled sampler scaffold
Mister-Raggs
committed
d873c85
[fix] exclude Cosmos training counters from conversion
Mister-Raggs
committed
c1c1368
[test] add Cosmos 2.5 distilled DiT parity gate
Mister-Raggs
committed
8cb88cd
[test] bypass Cosmos package CUDA-extra guard
Mister-Raggs
committed
24ce13c
test: isolate Cosmos DiT parity from config deps
Mister-Raggs
committed
c133880
test: run Cosmos reference DiT without transformer engine
Mister-Raggs
committed
f480f8d
test: complete reference RMSNorm interface
Mister-Raggs
committed
257bbe2
test: ignore Cosmos reference training counters
Mister-Raggs
committed
a704c29
test: trace Cosmos distilled DiT parity drift
Mister-Raggs
committed
613880b
test: calibrate Cosmos BF16 parity tolerance
Mister-Raggs
committed
a5acf0e
[Model] wire Cosmos2.5 distilled T2W inference
Mister-Raggs
committed
83665f9
[Test] match Cosmos distilled mask dtype
Mister-Raggs
committed
b023901
[bugfix] reason1: handle BatchEncoding from apply_chat_template
Mister-Raggs
committed
c5e5652
reason1: fail loudly on batch>1 chat_template output
Mister-Raggs
committed
84ab9de
[Test] add Cosmos distilled frame return gate
Mister-Raggs
committed
dbf65bf
[Docs] record Cosmos 2.5 distilled validation
Mister-Raggs
committed
72f25e4