fix: resolve Chapter 12 notebook compatibility with PyTorch 2.6.0 by specifying triton package version (#45)

This commit is contained in:
kvpratama
2025-04-04 15:25:35 +09:00
committed by GitHub
parent 361a342820
commit e55a0738aa

View File

@@ -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"
]
},
{