0e92ede59e
The Colab no-venv path in setup.sh (added in #4601, modified in #4618) only installs studio.txt (fastapi, uvicorn, structlog, etc.) into system Python. It skips install_python_stack entirely via _SKIP_PYTHON_DEPS=true, which means base.txt (unsloth + unsloth-zoo) and extras.txt (trl, transformers, etc.) are never installed. When the training worker subprocess spawns and tries to import unsloth, unsloth/__init__.py calls importlib.metadata.version("unsloth_zoo") which raises PackageNotFoundError, surfaced as: "Failed to import ML libraries: Please install unsloth_zoo" Fix: install base.txt and extras.txt in the Colab block, right after studio.txt. extras.txt gets the same version-constraint stripping as studio.txt to avoid conflicts with Colab's pre-installed packages. base.txt has no version pins so it is installed as-is. Only the Colab code path is changed. Non-Colab paths are untouched. |
||
|---|---|---|
| .. | ||
| backend | ||
| frontend | ||
| __init__.py | ||
| install_llama_prebuilt.py | ||
| install_python_stack.py | ||
| LICENSE.AGPL-3.0 | ||
| setup.bat | ||
| setup.ps1 | ||
| setup.sh | ||
| Unsloth_Studio_Colab.ipynb | ||