diff --git a/chapter12/Chapter 12 - Fine-tuning Generation Models.ipynb b/chapter12/Chapter 12 - Fine-tuning Generation Models.ipynb index 7a3b58f..decc439 100644 --- a/chapter12/Chapter 12 - Fine-tuning Generation Models.ipynb +++ b/chapter12/Chapter 12 - Fine-tuning Generation Models.ipynb @@ -47,7 +47,7 @@ "outputs": [], "source": [ "# %%capture\n", - "# !pip install -q accelerate==0.31.0 peft==0.11.1 bitsandbytes==0.43.1 transformers==4.41.2 trl==0.9.4 sentencepiece==0.2.0" + "# !pip install -q accelerate==0.31.0 peft==0.11.1 bitsandbytes==0.43.1 transformers==4.41.2 trl==0.9.4 sentencepiece==0.2.0 triton==3.1.0" ] }, {