| # 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 | |