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 2026-05-18 06:30:03 -07:00
commit e1fe3be939
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pyproject.toml
View file

@ -25,7 +25,7 @@ classifiers = [
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"unsloth_zoo>=2026.5.1",
"unsloth_zoo>=2026.5.2",
"wheel>=0.42.0",
"packaging",
"torch>=2.4.0,<2.11.0",
@ -76,16 +76,18 @@ studio = [
"frontend/*.yaml",
"frontend/.git*",
"backend/requirements/**/*",
"backend/plugins/**/*",
"backend/core/data_recipe/oxc-validator/*.json",
"backend/core/data_recipe/oxc-validator/*.mjs",
]
[tool.setuptools.packages.find]
exclude = ["images*", "tests*", "kernels/moe*"]
include = ["unsloth*", "unsloth_cli*", "studio", "studio.backend*"]
exclude = ["images*", "tests*", "*.node_modules", "*.node_modules.*"]
[project.optional-dependencies]
huggingface = [
"unsloth_zoo>=2026.5.1",
"unsloth_zoo>=2026.5.2",
"wheel>=0.42.0",
"packaging",
"torchvision",
@ -473,7 +475,7 @@ colab-ampere-torch220 = [
"unsloth[flashattention]",
]
colab-new = [
"unsloth_zoo>=2026.5.1",
"unsloth_zoo>=2026.5.2",
"packaging",
"tyro",
"transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0",