diff --git a/.github/workflows/notebooks-ci.yml b/.github/workflows/notebooks-ci.yml index 3ac6fe185d..36cc286cb7 100644 --- a/.github/workflows/notebooks-ci.yml +++ b/.github/workflows/notebooks-ci.yml @@ -197,9 +197,13 @@ jobs: # Pin to the same versions update_all_notebooks.py installs in # generated notebooks. Keep these in lockstep with PIN_TRL / # PIN_TRANSFORMERS in unslothai/notebooks/update_all_notebooks.py. + # `triton` is added because unsloth/_gpu_init.py:232 does an + # unconditional `import triton`; the PyPI wheel installs cleanly + # on Linux x86_64 even without CUDA (same rationale as + # consolidated-tests-ci.yml line 192-198). pip install 'transformers>=4.56,<5.6' 'trl>=0.22,<0.26' 'accelerate>=1.0' \ 'datasets>=3.4,<5' 'peft>=0.15,<0.20' \ - 'bitsandbytes>=0.43' 'sentencepiece' 'protobuf' + 'bitsandbytes>=0.43' 'sentencepiece' 'protobuf' triton # Converter deps (nbformat for notebook_to_python.py). pip install 'nbformat>=5.10' 'nbconvert>=7.16' pip install --no-deps unsloth_zoo unsloth