Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
1f2fbaa
Add Docker build CI workflow (#73)
xlu451 Jul 2, 2026
61460b6
Release: sync from i4 (vlm → reasoner rename) + inference/config fixe…
yy-code-nv Jul 2, 2026
9b02fa8
[Cosmos3 OSS]Add more action datasets (#72)
lzhang57 Jul 2, 2026
33ea6a7
Add Cosmos3-Nano LIBERO-10 action-policy SFT recipe, config, eval har…
fwd4 Jul 2, 2026
90290a2
Remove eval CLI references (#5)
xlu451 Jul 2, 2026
ea63366
Add multi-control transfer inference smoke test (#77)
lfengad Jul 2, 2026
fbb5c9b
Lazily import `lerobot` in datasets. (#76)
james77777778 Jul 2, 2026
df864d8
Release: vfm → generator, vlm → reasoner dest rename (moves + import …
yy-code-nv Jul 2, 2026
6efeea3
Convert Cosmos3-Nano vision tower in reasoner VLM export + fix public…
lfengad Jul 2, 2026
7127c4d
[Cosmos3 OSS] Fix RoboMind Dataset (#81)
lzhang57 Jul 3, 2026
0215967
Fix multi-control transfer: honor control weights under torch.compile…
trungtpham Jul 3, 2026
576bb79
Release: sync from internal (clean run) (#83)
yy-code-nv Jul 3, 2026
dff6966
deps: use natten gb300 (sm_103) wheel for x86_64 cu130 (#84)
pengcuo Jul 3, 2026
463ac14
Allow config aliases as public targets (#4)
xlu451 Jul 6, 2026
8139482
Release 2026-07-07 (i4 c2c7152e) (#91)
yy-code-nv Jul 7, 2026
8ea4318
DROID action-policy post-training: lazy LeRobot dataset + Cosmos3-Nan…
fwd4 Jul 7, 2026
3017efb
toml_config: add [job].upload_reproducible_setup, default False for O…
foreverlms Jul 8, 2026
b2237fd
convert_model_to_diffusers: export via official diffusers, drop diffu…
Dinghow Jul 9, 2026
6d84808
Support reasoner video input. (#25)
foreverlms Jul 9, 2026
3d9c087
Add Cosmos3-Super VideoPhy-2 Reasoner SFT recipe (#101)
Xuanmeng-Zhang Jul 10, 2026
1e139fb
feat(transfer-sft): control-conditioned SFT dataset + curator support…
lfengad Jul 11, 2026
0fa3ba4
Release 2026-07-11 (i4 7392c4b5) (#105)
yy-code-nv Jul 11, 2026
755bff4
Release 2026-07-13 (i4 c0d7b004) (#108)
yy-code-nv Jul 14, 2026
ace6140
feat: wire MXFP8/NVFP4 quantization into inference CLI (#113)
lfengad Jul 15, 2026
bc44b1e
Sync Cosmos3 Diffusers converter support (#110)
lfengad Jul 15, 2026
692c410
Release 2026-07-14 (i4 b66cfd81) (#109)
yy-code-nv Jul 15, 2026
acfe95c
Sync origin/main into kunlun through 692c410583a2e1271cdc355ffe2ffe57…
chaonanD Jul 16, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
483 changes: 483 additions & 0 deletions .file_mapping.json

Large diffs are not rendered by default.

28 changes: 21 additions & 7 deletions .github/workflows/gpu-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
# * training-smoke — Nano SFT pipeline (convert -> train 5 -> export -> t2i)
# * generator-training-regression — vision_sft_nano loss vs goldens (4-GPU subset)
# * generator-inference-smoke — Nano multi-modality inference (t2vs + policy + forward_dynamics)
# * reasoner-inference-smoke — Nano reasoner inference first-token logits vs golden (image-conditioned, 4-GPU)
# * reasoner-inference-smoke — Nano reasoner inference golden + Qwen conversion coverage (4-GPU)
# * reasoner-training-regression — llava_ov loss vs goldens (4-GPU subset)
#
# Requires:
Expand Down Expand Up @@ -76,7 +76,7 @@ jobs:
generator-training-regression:
needs: pre-commit
runs-on: [self-hosted, gpu, h200]
timeout-minutes: 60
timeout-minutes: 75
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
HF_HUB_DISABLE_XET: "1"
Expand All @@ -90,12 +90,14 @@ jobs:
- name: Sync environment (cu128-train)
run: uv sync --all-extras --group=cu128-train

# Generator (vision_sft_nano) loss vs the h100 goldens. -s streams the live log.
- name: Generator regression (vision_sft_nano, 4-GPU subset)
# Generator (vision_sft_nano) loss vs the h100 goldens, plus a Nano DCP ->
# Diffusers conversion that validates the exported component files and indexes.
# Both tests share the module-scoped staged Nano DCP. -s streams the live log.
- name: Generator regression and Nano Diffusers export (4-GPU subset)
run: |
export LD_LIBRARY_PATH=
uv run --all-extras --group=cu128-train python -m pytest -v -s \
tests/launch_regression_test.py -k vision_sft_nano \
tests/launch_regression_test.py -k 'vision_sft_nano or convert_model_to_diffusers' \
--num-gpus=4 --levels=2 -o addopts=

# The h100_inputs fixture removes its DCP stage on teardown; clear the
Expand Down Expand Up @@ -169,8 +171,20 @@ jobs:
tests/nano_reasoner_inference_smoke_test.py::test_nano_reasoner_image_first_token_logits \
--num-gpus=4 --levels=2 -o addopts=

# Reasoner inference writes only the pytest tmp dir (generated text + logs);
# the checkpoint download stays in the HF cache (kept).
# Merge Cosmos3-Nano into the public Qwen3-VL shell and verify that every
# Qwen tensor is present and both the vision and language towers come from
# Cosmos3-Nano. The exact node ID avoids collecting the training regressions
# in the same file.
- name: Nano reasoner Qwen tensor conversion coverage (4 GPU collection)
run: |
export LD_LIBRARY_PATH=
uv run --all-extras --group=cu128-train python -m pytest -v -s \
tests/launch_regression_test.py::test_convert_reasoner_converts_all_qwen_tensors \
--num-gpus=4 --levels=2 -o addopts=

# Reasoner inference and conversion write to pytest tmp dirs (generated
# text, logs, and the merged checkpoint); downloaded checkpoints stay in
# the HF cache (kept).
- name: Clean up run outputs
if: always()
run: |
Expand Down
Loading