setup.sh

#!/bin/bash

# Install torch first
pip install torch==2.4.1

# Then install the rest of the dependencies
pip install -r requirements.txt