Skip to content

Add part 2 of end-to-end tutorial: fine-tuning #2394

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

Merged
merged 1 commit into from
Jun 18, 2025
Merged

Conversation

andrewor14
Copy link
Contributor

@andrewor14 andrewor14 commented Jun 17, 2025

This commit adds the QAT tutorial and a general structure for the fine-tuning tutorial, which all also include QLoRA and float8 quantized fine-tuning. It also connects the 3 tutorial parts (pre-training, fine-tuning, and serving) into one cohesive end-to-end flow with some visuals and text.

Preview it yourself here: https://docs-preview.pytorch.org/pytorch/ao/2394/finetuning.html

Screenshot 2025-06-17 at 5 35 18 PM

Copy link

pytorch-bot bot commented Jun 17, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/2394

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 2c50f0a with merge base 8b12ddf (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@andrewor14 andrewor14 added the topic: documentation Use this tag if this PR adds or improves documentation label Jun 17, 2025
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jun 17, 2025
@andrewor14 andrewor14 force-pushed the finetuning-docs branch 2 times, most recently from 744d1f2 to e104d85 Compare June 17, 2025 21:45
@jerryzh168
Copy link
Contributor

thanks, the same image should probably be added to other tutorials as well

3. Use our integratino with `Axolotl <https://github.com/axolotl-ai-cloud/axolotl>`__


Option 1: TorchAO QAT API
Copy link
Contributor

Choose a reason for hiding this comment

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

when do we want people use this?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

I think if people have their own fine-tuning frameworks they want to use, e.g. the Llama-3.2 1B/3B quantized release did not use torchtune or axolotl

Copy link
Contributor

Choose a reason for hiding this comment

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

nit: I think we can order these from most popular to least popular

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Ok, I will put the integrations first since we want OSS users to use those first

@andrewor14
Copy link
Contributor Author

thanks, the same image should probably be added to other tutorials as well

Yeah I added the corresponding images in part 1 and part 3

@andrewor14 andrewor14 force-pushed the finetuning-docs branch 5 times, most recently from 7bcf5a6 to 65bb85c Compare June 18, 2025 13:51
This commit adds the QAT tutorial and a general structure for
the fine-tuning tutorial, which all also include QLoRA and float8
quantized fine-tuning. It also connects the 3 tutorial parts
(pre-training, fine-tuning, and serving) into one cohesive
end-to-end flow with some visuals and text.
@andrewor14 andrewor14 merged commit c561d26 into main Jun 18, 2025
22 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: documentation Use this tag if this PR adds or improves documentation
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants