-
Notifications
You must be signed in to change notification settings - Fork 440
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
config changes #1733
config changes #1733
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchtune/1733
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 9a94468 with merge base 55b4814 (): This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@@ -14,6 +14,12 @@ | |||
# tune run --nproc_per_node 8 full_finetune_distributed --config llama3/70B_full checkpointer.checkpoint_dir=<YOUR_CHECKPOINT_DIR> | |||
# | |||
# This config is only tested on an 8xA100 machine. | |||
# | |||
# !!!!!!!!!!!!! | |||
# !!!!!!!!!!!!! |
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.
nit: I would take the !!!! out personally
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.
i am afraid that users would ignore it without the "!", and not understand why it crashes
checkpoint_files: [ | ||
hf_model_0001_0.pt | ||
model.safetensors | ||
] | ||
recipe_checkpoint: null | ||
output_dir: /tmp/Qwen2-1.5B-Instruct-lora-finetune |
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.
should change output dir 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.
i thought about it. Indeed in all of our configs we use the same output dir. I will make the change
Co-authored-by: Felipe Mello <felipemello@fb.com>
Context
What is the purpose of this PR? Is it to
Changelog
3 mains fixes:
extra:
added compile flag for llama. I should have done it in another PR, sorry/not sorry.
Test plan
ran the configs and saved ckpt