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

Update pyproject.toml

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-25 03:55:26 +11:00
commit 0474451a6a
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pyproject.toml
View file

@ -98,10 +98,14 @@ colab_ampere = [
"flash-attn",
]
colab_torch211 = [
"unsloth[cu121_torch211]",
"unsloth[huggingface]",
"bitsandbytes",
"unsloth[cu121onlytorch211]",
]
colab_ampere_torch211 = [
"unsloth[cu121_torch211]",
"unsloth[huggingface]",
"bitsandbytes",
"unsloth[cu121onlytorch211]",
"packaging",
"ninja",
"flash-attn",