#!/bin/bash
# Upgrade pip
pip install --upgrade pip

# Force install the exact PyTorch version (CPU or CUDA as needed)
pip install --force-reinstall torch==2.1.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# Force reinstall Transformers and Sentence-Transformers to match PyTorch
pip install --force-reinstall transformers>=4.40.0,<5.0.0
pip install --force-reinstall sentence-transformers>=2.6.1,<3.0.0