Add Cosmos3-Super inference support to transfer cookbook#206
Open
trungtpham wants to merge 1 commit into
Open
Conversation
trungtpham
commented
Jun 11, 2026
Contributor
- Update notebook to support both Nano (single GPU) and Super (multi-GPU, 32B) via COSMOS3_MODEL env-var; consolidate §9-13 inference cells with if/else bash logic that switches launcher (python vs torchrun) and checkpoint path accordingly.
- Add user-editable config cell for HF_TOKEN, cache root, and output root; auto-detect available GPUs when COSMOS3_NUM_GPUS is unset.
- Route outputs to model-namespaced sub-dirs (e.g. outputs/.../Cosmos3-Super/) to prevent Nano and Super results from overwriting each other.
- Update preview_helpers.py to resolve output path from COSMOS3_MODEL.
- Update README with Super quickstart commands, model comparison table, and notebook usage instructions.
- Add .gitignore to exclude generated previews and outputs/.
c91a838 to
4635682
Compare
Restructure the transfer notebook to follow the audiovisual cookbook pattern — dedicated sections for each model instead of a single if/else-gated flag: - §9–§13 Cosmos3-Nano: python launcher, latency preset, single GPU - §14–§18 Cosmos3-Super: torchrun launcher, throughput preset, multi-GPU Other changes: - COSMOS3_NUM_GPUS auto-detects available GPUs (no hardcoded default). - CUDA_VISIBLE_DEVICES defaults to all detected GPUs so both model sections work without reconfiguration. - Each preview cell passes model= explicitly (no env-var dependency). - Update preview_helpers.py to route outputs to model-namespaced dirs (Cosmos3-Nano/ vs Cosmos3-Super/). - Update README with Super quickstart, model comparison table, and notebook instructions. - Add .gitignore to exclude generated previews and outputs/. - Clear cell outputs.
4635682 to
925d94c
Compare
lfengad
reviewed
Jun 12, 2026
|
|
||
| *model* selects which output directory to read (``Cosmos3-Nano`` uses | ||
| ``<output_root>/<control>/…``; ``Cosmos3-Super`` uses | ||
| ``<output_root>/<control>_super/…``). Defaults to the |
Collaborator
There was a problem hiding this comment.
Is this name consistent with the name in the code? seems no <control>_super such pattern found in the paths used in the code?
fferroni
approved these changes
Jun 12, 2026
Contributor
Author
|
@lfengad please help merge the PR if there is no more concerns. thanks |
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.