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:
Daniel Han 2026-04-02 18:02:59 +00:00
commit a484032f92
4 changed files with 45 additions and 19 deletions

View file

@ -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",