mirror of
https://github.com/allaunthefox/Research-Stack.git
synced 2026-07-31 03:05:21 +00:00
16 lines
589 B
Bash
16 lines
589 B
Bash
#!/bin/bash
|
|
# Install Unsloth and dependencies for NVIDIA CUDA
|
|
set -e
|
|
|
|
echo "Starting Unsloth installation for Gemma 4..."
|
|
|
|
# 1. Update pip
|
|
python3 -m pip install --upgrade pip
|
|
|
|
# 2. Install Torch and dependencies (Unsloth recommended versions)
|
|
# Note: Gemma 4 support usually requires latest unsloth
|
|
python3 -m pip install --upgrade "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
|
|
python3 -m pip install --no-deps "xformers<0.0.29" "trl<0.13.0" peft accelerate bitsandbytes
|
|
|
|
echo "Unsloth installation complete."
|
|
echo "You can now use Unsloth to load Gemma 4 models."
|