[doc] Add QAT tutorial

pytorch · Jun 22, 2024 · 44a2b4b · 44a2b4b
1 parent ef6e196
commit 44a2b4b
Show file tree

Hide file tree

Showing 3 changed files with 397 additions and 0 deletions.
diff --git a/docs/source/_static/img/qat_diagram.png b/docs/source/_static/img/qat_diagram.png
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -64,6 +64,13 @@ torchtune tutorials.
    :link: tutorials/qlora_finetune.html
    :tags: finetuning,llama2,qlora
 
+.. customcarditem::
+   :header: Finetuning with QAT in torchtune
+   :card_description: Finetuning of Llama3 using QAT
+   :image: _static/img/generic-pytorch-logo.png
+   :link: tutorials/qat_finetune.html
+   :tags: finetuning,llama3,qat,quantization,evals
+
 .. customcarditem::
    :header: End-to-End Workflow with torchtune
    :card_description: Train, Evaluate, Quantize and then Generate with your LLM.
@@ -99,6 +106,7 @@ torchtune tutorials.
    tutorials/llama3
    tutorials/lora_finetune
    tutorials/qlora_finetune
+   tutorials/qat_finetune
    tutorials/e2e_flow
    tutorials/datasets
    tutorials/chat