From e55a0738aa56a16a408737ab9ae98a7dd94f54ad Mon Sep 17 00:00:00 2001 From: kvpratama Date: Fri, 4 Apr 2025 15:25:35 +0900 Subject: [PATCH] fix: resolve Chapter 12 notebook compatibility with PyTorch 2.6.0 by specifying triton package version (#45) --- chapter12/Chapter 12 - Fine-tuning Generation Models.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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" ] }, {