Skip to content

Release 2026-07-14 (i4 97901d3959) #131

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
yy-code-nv wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Release 2026-07-14 (i4 97901d3959) #131

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
e5a051c
Release 2026-07-14 (i4 97901d3959)
Jul 23, 2026
b43e5dd
Fix GPU CI regressions from the 2026-07-14 sync
lfengad Jul 23, 2026
4818316
Merge branch 'main' into release/2026-07-14-97901d3959
lfengad Jul 23, 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
43 changes: 40 additions & 3 deletions .file_mapping.json
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
{
"_source_commit": "b66cfd8155158744e16d598b7515fbd0b5081a24-dirty",
"_dest_commit": "755bff48dd66b467995366796e747ffc649dd83c",
"_generated_at": "2026-07-14T08:42:29Z",
"_source_commit": "97901d395925c227163a7370ec2763b4057783d5",
"_dest_commit": "058c8c046877f087e07b3472a76cac214f3ad39c",
"_generated_at": "2026-07-23T11:16:20Z",
"files": {
"imaginaire/__init__.py": "cosmos_framework/__init__.py",
"imaginaire/attention/__init__.py": "cosmos_framework/model/attention/__init__.py",
Expand Down Expand Up @@ -98,13 +98,15 @@
"imaginaire/lazy_config/instantiate.py": "cosmos_framework/utils/lazy_config/instantiate.py",
"imaginaire/lazy_config/lazy.py": "cosmos_framework/utils/lazy_config/lazy.py",
"imaginaire/lazy_config/lazy_call.py": "cosmos_framework/utils/lazy_config/lazy_call.py",
"imaginaire/lazy_config/lazy_test.py": "cosmos_framework/utils/lazy_config/lazy_test.py",
"imaginaire/lazy_config/omegaconf_patch.py": "cosmos_framework/utils/lazy_config/omegaconf_patch.py",
"imaginaire/lazy_config/registry.py": "cosmos_framework/utils/lazy_config/registry.py",
"imaginaire/model.py": "cosmos_framework/model/_base.py",
"imaginaire/serialization.py": "cosmos_framework/utils/serialization.py",
"imaginaire/trainer.py": "cosmos_framework/trainer/__init__.py",
"imaginaire/utils/__init__.py": "cosmos_framework/utils/__init__.py",
"imaginaire/utils/callback.py": "cosmos_framework/utils/callback.py",
"imaginaire/utils/callback_test.py": "cosmos_framework/utils/callback_test.py",
"imaginaire/utils/checkpoint_db.py": "cosmos_framework/utils/checkpoint_db.py",
"imaginaire/utils/checkpointer.py": "cosmos_framework/utils/checkpointer.py",
"imaginaire/utils/cluster_env.py": "cosmos_framework/utils/cluster_env.py",
Expand All @@ -113,6 +115,7 @@
"imaginaire/utils/count_params.py": "cosmos_framework/utils/count_params.py",
"imaginaire/utils/device.py": "cosmos_framework/utils/device.py",
"imaginaire/utils/distributed.py": "cosmos_framework/utils/distributed.py",
"imaginaire/utils/distributed_test.py": "cosmos_framework/utils/distributed_test.py",
"imaginaire/utils/easy_io/README.md": "cosmos_framework/utils/easy_io/README.md",
"imaginaire/utils/easy_io/__init__.py": "cosmos_framework/utils/easy_io/__init__.py",
"imaginaire/utils/easy_io/backends/__init__.py": "cosmos_framework/utils/easy_io/backends/__init__.py",
Expand Down Expand Up @@ -243,6 +246,8 @@
"projects/cosmos3/cosmos3/configs/base/reasoner/experiment/__init__.py": "cosmos_framework/configs/base/reasoner/experiment/__init__.py",
"projects/cosmos3/cosmos3/configs/base/reasoner/experiment/utils.py": "cosmos_framework/configs/base/reasoner/experiment/utils.py",
"projects/cosmos3/cosmos3/configs/base/reasoner/freeze_config.py": "cosmos_framework/configs/base/reasoner/freeze_config.py",
"projects/cosmos3/cosmos3/configs/distillation_export.py": "cosmos_framework/inference/common/distillation_export.py",
"projects/cosmos3/cosmos3/configs/distillation_export_test.py": "cosmos_framework/inference/common/distillation_export_test.py",
"projects/cosmos3/cosmos3/datasets/action/__init__.py": "cosmos_framework/data/generator/action/__init__.py",
"projects/cosmos3/cosmos3/datasets/action/action_processing.py": "cosmos_framework/data/generator/action/action_processing.py",
"projects/cosmos3/cosmos3/datasets/action/action_spec.py": "cosmos_framework/data/generator/action/action_spec.py",
Expand Down Expand Up @@ -302,6 +307,7 @@
"projects/cosmos3/cosmos3/datasets/augmentors/transfer_control_input/seg.py": "cosmos_framework/data/generator/augmentors/transfer_control_input/seg.py",
"projects/cosmos3/cosmos3/datasets/augmentors/transfer_control_transform.py": "cosmos_framework/data/generator/augmentors/transfer_control_transform.py",
"projects/cosmos3/cosmos3/datasets/augmentors/video_parsing.py": "cosmos_framework/data/generator/augmentors/video_parsing.py",
"projects/cosmos3/cosmos3/datasets/augmentors/video_parsing_test.py": "cosmos_framework/data/generator/augmentors/video_parsing_test.py",
"projects/cosmos3/cosmos3/datasets/joint_dataloader.py": "cosmos_framework/data/generator/joint_dataloader.py",
"projects/cosmos3/cosmos3/datasets/packing_iterable_dataset.py": "cosmos_framework/data/generator/packing_iterable_dataset.py",
"projects/cosmos3/cosmos3/datasets/reasoner/video_decoder_qwen.py": "cosmos_framework/data/generator/reasoner/video_decoder_qwen.py",
Expand All @@ -328,18 +334,25 @@
"projects/cosmos3/cosmos3/models/mot/cosmos3_vfm_network.py": "cosmos_framework/model/generator/mot/cosmos3_vfm_network.py",
"projects/cosmos3/cosmos3/models/mot/domain_aware_linear.py": "cosmos_framework/model/generator/mot/domain_aware_linear.py",
"projects/cosmos3/cosmos3/models/mot/dot_product_attention.py": "cosmos_framework/model/generator/mot/dot_product_attention.py",
"projects/cosmos3/cosmos3/models/mot/inference_text_kv_memory.py": "cosmos_framework/model/generator/mot/inference_text_kv_memory.py",
"projects/cosmos3/cosmos3/models/mot/modeling_utils.py": "cosmos_framework/model/generator/mot/modeling_utils.py",
"projects/cosmos3/cosmos3/models/mot/parallelize_unified_mot.py": "cosmos_framework/model/generator/mot/parallelize_unified_mot.py",
"projects/cosmos3/cosmos3/models/mot/parallelize_vfm_network.py": "cosmos_framework/model/generator/mot/parallelize_vfm_network.py",
"projects/cosmos3/cosmos3/models/mot/und_k_norm_example_test.py": "cosmos_framework/model/generator/mot/und_k_norm_example_test.py",
"projects/cosmos3/cosmos3/models/mot/unified_mot.py": "cosmos_framework/model/generator/mot/unified_mot.py",
"projects/cosmos3/cosmos3/models/mot/unified_mot_test.py": "cosmos_framework/model/generator/mot/unified_mot_test.py",
"projects/cosmos3/cosmos3/models/omni_mot_model.py": "cosmos_framework/model/generator/omni_mot_model.py",
"projects/cosmos3/cosmos3/models/parallelize_vlm.py": "cosmos_framework/model/generator/parallelize_vlm.py",
"projects/cosmos3/cosmos3/models/reasoner/__init__.py": "cosmos_framework/model/generator/reasoner/__init__.py",
"projects/cosmos3/cosmos3/models/reasoner/cosmos3_edge/__init__.py": "cosmos_framework/model/generator/reasoner/cosmos3_edge/__init__.py",
"projects/cosmos3/cosmos3/models/reasoner/cosmos3_edge/configuration_cosmos3_edge.py": "cosmos_framework/model/generator/reasoner/cosmos3_edge/configuration_cosmos3_edge.py",
"projects/cosmos3/cosmos3/models/reasoner/cosmos3_edge/modeling_cosmos3_edge.py": "cosmos_framework/model/generator/reasoner/cosmos3_edge/modeling_cosmos3_edge.py",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/__init__.py": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/__init__.py",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/configs/Nemotron-2B-Dense-VL.json": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/configs/Nemotron-2B-Dense-VL.json",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/configuration_nemotron_3_dense_vl.py": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/configuration_nemotron_3_dense_vl.py",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/nemotron_3_dense_vl.py": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/nemotron_3_dense_vl.py",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/reasoner_multimodal_utils.py": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/reasoner_multimodal_utils.py",
"projects/cosmos3/cosmos3/models/reasoner/nemotron_3_dense_vl/vision_siglip2.py": "cosmos_framework/model/generator/reasoner/nemotron_3_dense_vl/vision_siglip2.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl/__init__.py": "cosmos_framework/model/generator/reasoner/qwen3_vl/__init__.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl/configs/Qwen3-VL-2B-Instruct.json": "cosmos_framework/model/generator/reasoner/qwen3_vl/configs/Qwen3-VL-2B-Instruct.json",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl/configs/Qwen3-VL-32B-Instruct.json": "cosmos_framework/model/generator/reasoner/qwen3_vl/configs/Qwen3-VL-32B-Instruct.json",
Expand All @@ -355,6 +368,7 @@
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/configs/Qwen3-VL-30B-A3B-Instruct.json": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/configs/Qwen3-VL-30B-A3B-Instruct.json",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/configs/__init__.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/configs/__init__.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/configuration_qwen3_vl_moe.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/configuration_qwen3_vl_moe.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/cosine_router_test.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/cosine_router_test.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/moe.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/moe.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/moe_bench.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/moe_bench.py",
"projects/cosmos3/cosmos3/models/reasoner/qwen3_vl_moe/moe_kernels.py": "cosmos_framework/model/generator/reasoner/qwen3_vl_moe/moe_kernels.py",
Expand All @@ -363,12 +377,16 @@
"projects/cosmos3/cosmos3/models/utils/__init__.py": "cosmos_framework/model/generator/utils/__init__.py",
"projects/cosmos3/cosmos3/models/utils/data_and_condition.py": "cosmos_framework/model/generator/utils/data_and_condition.py",
"projects/cosmos3/cosmos3/models/utils/memory.py": "cosmos_framework/model/generator/utils/memory.py",
"projects/cosmos3/cosmos3/models/utils/moe_utils.py": "cosmos_framework/model/generator/utils/moe_utils.py",
"projects/cosmos3/cosmos3/models/utils/safetensors_loader.py": "cosmos_framework/model/generator/utils/safetensors_loader.py",
"projects/cosmos3/cosmos3/models/utils/safetensors_loader_indexed_test.py": "cosmos_framework/model/generator/utils/safetensors_loader_indexed_test.py",
"projects/cosmos3/cosmos3/models/utils/safetensors_loader_test.py": "cosmos_framework/model/generator/utils/safetensors_loader_test.py",
"projects/cosmos3/cosmos3/models/utils/taylorseer.py": "cosmos_framework/model/generator/utils/taylorseer.py",
"projects/cosmos3/cosmos3/models/vlm_model.py": "cosmos_framework/model/generator/vlm_model.py",
"projects/cosmos3/cosmos3/processors/__init__.py": "cosmos_framework/data/generator/processors/__init__.py",
"projects/cosmos3/cosmos3/processors/base.py": "cosmos_framework/data/generator/processors/base.py",
"projects/cosmos3/cosmos3/processors/cosmos3_edge_processing.py": "cosmos_framework/data/generator/processors/cosmos3_edge_processing.py",
"projects/cosmos3/cosmos3/processors/cosmos3_edge_processing_test.py": "cosmos_framework/data/generator/processors/cosmos3_edge_processing_test.py",
"projects/cosmos3/cosmos3/processors/nemotron3densevl_processor.py": "cosmos_framework/data/generator/processors/nemotron3densevl_processor.py",
"projects/cosmos3/cosmos3/processors/nemotronvl_processor.py": "cosmos_framework/data/generator/processors/nemotronvl_processor.py",
"projects/cosmos3/cosmos3/processors/qwen3vl_processor.py": "cosmos_framework/data/generator/processors/qwen3vl_processor.py",
Expand Down Expand Up @@ -435,7 +453,26 @@
"projects/cosmos3/cosmos3/utils/reasoner/flop_calculator.py": "cosmos_framework/utils/generator/reasoner/flop_calculator.py",
"projects/cosmos3/cosmos3/utils/reasoner/pretrained_models_downloader.py": "cosmos_framework/utils/generator/reasoner/pretrained_models_downloader.py",
"projects/cosmos3/cosmos3/utils/video_preprocess.py": "cosmos_framework/utils/generator/video_preprocess.py",
"projects/cosmos3/interactive/callbacks/dmd2_parity_ledger.py": "cosmos_framework/callbacks/dmd2_metrics.py",
"projects/cosmos3/interactive/callbacks/dmd2_parity_ledger_test.py": "cosmos_framework/callbacks/dmd2_metrics_test.py",
"projects/cosmos3/interactive/callbacks/grad_clip_cosmos3.py": "cosmos_framework/callbacks/grad_clip_distillation.py",
"projects/cosmos3/interactive/callbacks/grad_clip_cosmos3_test.py": "cosmos_framework/callbacks/grad_clip_distillation_test.py",
"projects/cosmos3/interactive/checkpointer/dcp_distill_cosmos3.py": "cosmos_framework/checkpoint/dcp_distill.py",
"projects/cosmos3/interactive/checkpointer/dcp_distill_cosmos3_test.py": "cosmos_framework/checkpoint/dcp_distill_test.py",
"projects/cosmos3/interactive/configs/methods/__init__.py": "cosmos_framework/configs/base/experiment/distillation/__init__.py",
"projects/cosmos3/interactive/configs/methods/dmd2.py": "cosmos_framework/configs/base/experiment/distillation/dmd2_config.py",
"projects/cosmos3/interactive/configs/methods/dmd2_test.py": "cosmos_framework/configs/base/experiment/distillation/dmd2_config_test.py",
"projects/cosmos3/interactive/methods/common_loss.py": "cosmos_framework/model/generator/distillation/common_loss.py",
"projects/cosmos3/interactive/methods/common_loss_test.py": "cosmos_framework/model/generator/distillation/common_loss_test.py",
"projects/cosmos3/interactive/methods/distribution_matching/__init__.py": "cosmos_framework/model/generator/distillation/__init__.py",
"projects/cosmos3/interactive/methods/distribution_matching/dmd2_rf.py": "cosmos_framework/model/generator/distillation/dmd2_rf.py",
"projects/cosmos3/interactive/methods/distribution_matching/dmd2_rf_test.py": "cosmos_framework/model/generator/distillation/dmd2_rf_test.py",
"projects/cosmos3/interactive/optimizer.py": "cosmos_framework/model/generator/distillation/optimizer.py",
"projects/cosmos3/interactive/optimizer_test.py": "cosmos_framework/model/generator/distillation/optimizer_test.py",
"projects/cosmos3/interactive/trainer.py": "cosmos_framework/trainer/distillation.py",
"projects/cosmos3/interactive/trainer_test.py": "cosmos_framework/trainer/distillation_test.py",
"projects/cosmos3/tokenizer/checkpoint_io.py": "cosmos_framework/model/tokenizer/checkpoint_io.py",
"projects/cosmos3/tokenizer/evaluation/lpips_cache.py": "cosmos_framework/model/tokenizer/evaluation/lpips_cache.py",
"projects/cosmos3/tokenizer/evaluation/metric.py": "cosmos_framework/model/tokenizer/evaluation/metric.py",
"projects/cosmos3/tokenizer/evaluation/reconstruction_metrics.py": "cosmos_framework/model/tokenizer/evaluation/reconstruction_metrics.py",
"projects/cosmos3/tokenizer/models/__init__.py": "cosmos_framework/model/tokenizer/models/__init__.py",
Expand Down
3 changes: 0 additions & 3 deletions cosmos_framework/callbacks/dmd2_metrics.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,11 +24,8 @@
PARITY_KEYS: tuple[str, ...] = (
"vsd_loss",
"total_generator_loss",
"dmd_loss_generator",
"dmd_vsd_grad_norm",
"fake_score_loss",
"total_critic_loss",
"dmd_loss_critic",
"clip_grad_norm/net_selected_preclip",
"clip_grad_norm/net_selected_clip_scale",
"clip_grad_norm/net_selected_clip_norm",
Expand Down
Loading