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

pip: allow torch<2.12 and add amd/huggingfacenotorch extras

Browse source 
The released torch>=2.4.0,<2.11.0 pin excludes the AMD 2.11.0+rocm wheel, so a
follow-up pip resolve downgrades torch to 2.10.0 and breaks the ROCm install
(#7275). Raise the ceiling to <2.12.0 so the ROCm 2.11 wheel is preserved, and
add an amd extra (pulls huggingfacenotorch, no torch pin) plus a torch-free
huggingfacenotorch extra so AMD users keep their ROCm torch.

[pre-commit.ci skip]
This commit is contained in:
Daniel Han 2026-07-21 02:44:55 -07:00 • committed by danielhanchen
commit cd50f9e51d
1 changed files with 26 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

27
pyproject.toml
View file

@ -28,7 +28,7 @@ dependencies = [
"unsloth_zoo>=2026.7.4",
"wheel>=0.42.0",
"packaging",
"torch>=2.4.0,<2.11.0",
"torch>=2.4.0,<2.12.0",
"torchvision",
"numpy",
"tqdm",
@ -115,6 +115,31 @@ huggingface = [
"pyyaml",
"nest-asyncio",
]
huggingfacenotorch = [
"unsloth_zoo>=2026.7.4",
"wheel>=0.42.0",
"packaging",
"numpy",
"tqdm",
"psutil",
"tyro",
"protobuf",
"sentencepiece>=0.2.0",
"datasets>=3.4.1,!=4.0.*,!=4.1.0,<4.4.0",
"accelerate>=0.34.1",
"peft>=0.18.0,!=0.11.0",
"huggingface_hub>=0.34.0",
"hf_transfer",
"diffusers",
"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",
"trl>=0.18.2,!=0.19.0,<=0.24.0",
"sentence-transformers",
]
amd = [
"unsloth[huggingfacenotorch]",
"bitsandbytes>=0.49.1 ; ('linux' in sys_platform) and (platform_machine == 'AMD64' or platform_machine == 'x86_64' or platform_machine == 'aarch64')",
"bitsandbytes>=0.49.1 ; (sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
]
windows = [
"unsloth[huggingface]",
"bitsandbytes>=0.45.5,!=0.46.0,!=0.48.0 ; (sys_platform == 'win32')",