-
Notifications
You must be signed in to change notification settings - Fork 44
[QEff Finetune]: Enable --help for finetune CLI #392
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
Conversation
Signed-off-by: Mamta Singh <[email protected]>
Signed-off-by: Mamta Singh <[email protected]>
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.
Overall good cleanup. Please check comments.
@@ -58,7 +59,8 @@ class TrainConfig: | |||
model_name: str = "meta-llama/Llama-3.2-1B" | |||
tokenizer_name: str = None # if not passed as an argument, it uses the value of model_name | |||
run_validation: bool = True | |||
batch_size_training: int = 1 | |||
train_batch_size: int = 1 |
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.
if we are changing this param, may be an internal announcement is required. SIT team's testing commands might depend on this one.
3128d6e
to
1ca1ab4
Compare
Signed-off-by: Mamta Singh <[email protected]>
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.
Overall good cleanup.
c9c4c6e
to
f529f21
Compare
68316b0
to
5079464
Compare
Signed-off-by: Mamta Singh <[email protected]>
Signed-off-by: Mamta Singh <[email protected]>
1. Enabled --help for finetune 2. Updated finetune docs 3. removed unused flags --------- Signed-off-by: Mamta Singh <[email protected]> Co-authored-by: Mamta Singh <[email protected]>
Uh oh!
There was an error while loading. Please reload this page.