fix(studio): temporarily install unsloth and unsloth_zoo from git main
The latest PyPI releases of unsloth and unsloth_zoo do not include recent fixes needed by Studio (Gemma 4 support, compiler patches). Switch all install paths to pull from git main until the next PyPI release is cut. Changes: - install.sh: add _DEFAULT_UNSLOTH_PKG / _DEFAULT_ZOO_PKG defaults, replace 7 hardcoded "unsloth>=2026.3.18" / "unsloth-zoo" references - install.ps1: same pattern with $DefaultUnslothPkg / $DefaultZooPkg - install_python_stack.py: add _DEFAULT_UNSLOTH_PKG / _DEFAULT_ZOO_PKG, update the no-torch install path - base.txt: switch from package names to git+https URLs To revert: change the defaults back to "unsloth>=2026.3.18" and "unsloth-zoo" in all four files and restore base.txt to package names.
This commit is contained in:
parent
1ce83c40aa
commit
a484032f92
4 changed files with 45 additions and 19 deletions
|
|
@ -42,6 +42,15 @@ def _infer_no_torch() -> bool:
|
|||
|
||||
NO_TORCH = _infer_no_torch()
|
||||
|
||||
# -- Maintainer-editable package defaults --------------------------------------
|
||||
# Temporarily install from git main instead of PyPI to pick up latest fixes
|
||||
# (Gemma 4 support, Studio patches). Revert to PyPI package names once the
|
||||
# next release is cut.
|
||||
# PyPI: _DEFAULT_UNSLOTH_PKG = "unsloth"
|
||||
# _DEFAULT_ZOO_PKG = "unsloth-zoo"
|
||||
_DEFAULT_UNSLOTH_PKG = "git+https://github.com/unslothai/unsloth.git"
|
||||
_DEFAULT_ZOO_PKG = "git+https://github.com/unslothai/unsloth_zoo.git"
|
||||
|
||||
# -- Verbosity control ----------------------------------------------------------
|
||||
# By default the installer shows a minimal progress bar (one line, in-place).
|
||||
# Set UNSLOTH_VERBOSE=1 in the environment to restore full per-step output:
|
||||
|
|
@ -471,11 +480,11 @@ def install_python_stack() -> int:
|
|||
"--no-cache-dir",
|
||||
"--no-deps",
|
||||
"--upgrade-package",
|
||||
package_name,
|
||||
"unsloth",
|
||||
"--upgrade-package",
|
||||
"unsloth-zoo",
|
||||
package_name,
|
||||
"unsloth-zoo",
|
||||
_DEFAULT_UNSLOTH_PKG,
|
||||
_DEFAULT_ZOO_PKG,
|
||||
)
|
||||
pip_install(
|
||||
"Installing no-torch runtime deps",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue