ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update Colab to CUDA 12.1

Browse source
This commit is contained in:
Daniel Han-Chen 2023-12-15 16:35:24 +11:00
commit 399f8ed56f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pyproject.toml
View file

@ -65,7 +65,7 @@ kaggle = [
"unsloth[huggingface]",
]
colab = [
"unsloth[cu118]",
"unsloth[cu121]",
]
cu118_ampere = [
"unsloth[huggingface]",
@ -82,7 +82,7 @@ cu121_ampere = [
"flash-attn",
]
colab_ampere = [
"unsloth[cu118]",
"unsloth[cu121]",
"ninja",
"flash-attn",
]