fix: resolve Chapter 12 notebook compatibility with PyTorch 2.6.0 by specifying triton package version (#45)
This commit is contained in:
@@ -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"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user