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

build: add opt-in gefenx extra for gefen-x (now on PyPI)

Browse source 
gefen-x 0.2.0 is published on PyPI, so users can install the Gefen-X
optimizer dependency via an opt-in extra instead of from source:
`pip install "unsloth[gefenx]"`. gefen-x is CUDA-only and lazily
imported, so the base install is unchanged.
This commit is contained in:
thad0ctor 2026-07-10 11:11:54 -07:00
commit 95dbb31cb5
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
pyproject.toml
View file

@ -106,6 +106,12 @@ windows = [
base = [
"unsloth[huggingface]",
]
# Gefen-X optimizer (gefenx / gefenx_muon). Opt-in extra — gefen-x is a
# CUDA-only, lazily-imported optional dependency, so the base install is
# untouched. See unsloth/optimizers/gefenx.py.
gefenx = [
"gefen-x>=0.2.0",
]
cu118only = [
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9' and ('linux' in sys_platform)",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl ; python_version=='3.10' and ('linux' in sys_platform)",