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-03-14 20:27:20 +11:00
commit 5c4241a240
1 changed files with 2 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

13
pyproject.toml
View file

@ -161,7 +161,7 @@ colab-ampere-torch220 = [
"ninja",
"flash-attn",
]
colab-1 = [
colab-new = [
"tyro",
"transformers>=4.38.2",
"datasets>=2.16.0",
@ -171,22 +171,13 @@ colab-1 = [
"wheel>=0.42.0",
"numpy",
]
colab-2 = [
colab-no-deps = [
"accelerate>=0.26.1",
"trl>=0.7.9",
"peft>=0.7.1",
"xformers",
"bitsandbytes",
]
colab-ampere-1 = [
"unsloth[colab-1]",
]
colab-ampere-2 = [
"packaging",
"ninja",
"flash-attn",
"unsloth[colab-2]",
]
colab-ampere = [
"unsloth[colab-ampere-torch220]",
"packaging",