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

Fix base.txt filter to match 'unsloth' exactly, not 'unsloth-zoo'

Browse source 
The previous _filter_requirements approach matched any line starting
with "unsloth", which also caught "unsloth-zoo". Replace with a regex
that matches "unsloth" followed by a version specifier or end-of-line,
preserving "unsloth-zoo" and other packages with the "unsloth-" prefix.
This commit is contained in:
Daniel Han 2026-04-03 13:52:30 +00:00
commit 2bcafd1c25
1 changed files with 22 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

27
studio/install_python_stack.py
View file

@ -534,19 +534,36 @@ def install_python_stack() -> int:
f"git+https://github.com/unslothai/unsloth.git@{_git_ref}",
constrain = False,
)
# Filter out "unsloth" from base.txt so pip does not
# revert the git install back to the PyPI wheel.
_base_no_unsloth = _filter_requirements(REQ_ROOT / "base.txt", {"unsloth"})
# Install remaining base deps (unsloth-zoo etc.) but exclude
# the "unsloth" line so pip does not revert the git checkout
# back to the PyPI wheel. We match "unsloth" exactly (not
# "unsloth-zoo") by checking for a version specifier or EOL
# immediately after the package name.
import re as _re
_base_lines = (REQ_ROOT / "base.txt").read_text(
encoding = "utf-8"
).splitlines(keepends = True)
_base_filtered = [
ln
for ln in _base_lines
if not _re.match(r"^\s*unsloth\s*([<>=!~;\[#]|$)", ln.strip())
]
_base_tmp = tempfile.NamedTemporaryFile(
mode = "w", suffix = ".txt", delete = False, encoding = "utf-8",
)
_base_tmp.writelines(_base_filtered)
_base_tmp.close()
try:
pip_install(
"Updating remaining base packages",
"--no-cache-dir",
"--upgrade-package",
"unsloth-zoo",
req = _base_no_unsloth,
req = Path(_base_tmp.name),
)
finally:
_base_no_unsloth.unlink(missing_ok = True)
Path(_base_tmp.name).unlink(missing_ok = True)
else:
# Update path: upgrade only unsloth + unsloth-zoo while preserving
# existing torch/CUDA installations. Torch is pre-installed by