install.sh

#!/bin/zsh

# Активация виртуального окружения
if [ -d "venv" ]; then
    source venv/bin/activate
elif [ -d ".venv" ]; then
    source .venv/bin/activate
else
    echo "Виртуальное окружение не найдено. Создайте его командой: python3 -m venv venv"
    exit 1
fi

# Установка PyTorch через официальный индекс
echo "Установка PyTorch..."
pip install torch --index-url https://download.pytorch.org/whl/cpu

# Установка остальных зависимостей
echo "Установка остальных зависимостей..."
pip install transformers peft gradio accelerate safetensors

# bitsandbytes опционален (может не работать на всех системах)
echo "Попытка установки bitsandbytes (опционально)..."
pip install bitsandbytes || echo "bitsandbytes не установлен (это нормально)"

echo "Установка завершена!"