From c3c4ea0bc3312e087cc4c2a244746251375931d8 Mon Sep 17 00:00:00 2001 From: Dmitrii Ioksha <41271764+DmitryIo@users.noreply.github.com> Date: Sun, 10 Mar 2024 21:41:45 +0300 Subject: [PATCH] [doc-fix] Add accelerate required for notebook 3/3 --- utils/generate_notebooks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/generate_notebooks.py b/utils/generate_notebooks.py index f4e77cd62..9ed74f976 100644 --- 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)"