[doc-fix] Add accelerate required for notebook 3/3

apointeau · Mar 10, 2024 · c3c4ea0 · c3c4ea0
1 parent e63a663
commit c3c4ea0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/generate_notebooks.py b/utils/generate_notebooks.py
@@ -131,6 +131,7 @@ def build_notebook(fname, title, output_dir="."):
     """
     sections = read_and_split_frameworks(fname)
     sections_with_accelerate = [
+        "chapter3/3",  # "Fine-tuning a model with the Trainer API or Keras",
         "chapter3/4",  # "A full training",
         "chapter7/2_pt",  # "Token classification (PyTorch)",
         "chapter7/3_pt",  # "Fine-tuning a masked language model (PyTorch)"