-
Notifications
You must be signed in to change notification settings - Fork 294
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.
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
Add more Float8 description #284
Conversation
train_configs/debug_model.toml
Outdated
@@ -37,7 +37,7 @@ steps = 10 | |||
data_parallel_degree = -1 | |||
tensor_parallel_degree = 1 | |||
pipeline_parallel_degree = 1 | |||
fp8_linear = "" | |||
fp8_linear = '' # ['', 'dynamic'] See torchtitan/float8_linear.py for more detail |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
IMO we should put details about a config into the helper message in config_manager.py
. Here we can simply have # ['', 'dynamic']
as the possible options, similar to the norm_type
config. But I'm ok with also putting them here.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yeah that makes sense, I was trying to find the best place for this one I can update this PR
8504e2c
to
5ce52a6
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm! thanks!
# Summary Add more the possible options in the configs and add a note on how to get the dependency at the top of the file.
# Summary Add more the possible options in the configs and add a note on how to get the dependency at the top of the file.
Summary
Add more the possible options in the configs and add a note on how to get the dependency at the top of the file.