Skip to content

Adding gpt_oss_20b and gpt_oss_120b to config_registry.py#2432

Merged
tianyu-l merged 3 commits intopytorch:mainfrom
chelsea0x3b:gpt-oss-configs
Feb 26, 2026
Merged

Adding gpt_oss_20b and gpt_oss_120b to config_registry.py#2432
tianyu-l merged 3 commits intopytorch:mainfrom
chelsea0x3b:gpt-oss-configs

Conversation

@chelsea0x3b
Copy link
Contributor

@chelsea0x3b chelsea0x3b commented Feb 24, 2026 •
edited
Loading

Tested with

torchrun --nproc-per-node gpu -m torchtitan.train --module gpt_oss --config gpt_oss_20b --training.local-batch-size 4

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Feb 24, 2026
tianyu-l
tianyu-l reviewed Feb 24, 2026
View reviewed changes
@chelsea0x3b chelsea0x3b requested a review from tianyu-l February 25, 2026 14:01
tianyu-l
tianyu-l approved these changes Feb 25, 2026
View reviewed changes
Copy link
Contributor

@tianyu-l tianyu-l left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@tianyu-l
Copy link
Contributor

tianyu-l commented Feb 25, 2026

please fix linting

@chelsea0x3b chelsea0x3b requested a review from tianyu-l February 26, 2026 17:27
tianyu-l
tianyu-l approved these changes Feb 26, 2026
View reviewed changes
Copy link
Contributor

@tianyu-l tianyu-l left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@tianyu-l tianyu-l merged commit 7464aef into pytorch:main Feb 26, 2026
9 of 11 checks passed
@chelsea0x3b chelsea0x3b deleted the gpt-oss-configs branch February 26, 2026 20:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@tianyu-l tianyu-l tianyu-l approved these changes

@fegin fegin Awaiting requested review from fegin fegin is a code owner

@wwwjn wwwjn Awaiting requested review from wwwjn wwwjn is a code owner

@wconstab wconstab Awaiting requested review from wconstab wconstab is a code owner

Assignees

No one assigned

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@chelsea0x3b @tianyu-l @qayqaq