Add accelerate required for notebook

[doc-fix] Add accelerate required for notebook 3/3
huggingface · May 17, 2024 · a91486c · a91486c
2 parents bb1c442 + c3c4ea0
commit a91486c
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)"