Update runpod

requirements_runpod.txt

@@ -1,4 +1,4 @@
-torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
+torch==2.0.1+cu117 torchvision==0.15.2+cu117 --extra-index-url https://download.pytorch.org/whl/cu117 # no_verify leave this to specify not checking this a verification stage
 xformers==0.0.20 bitsandbytes==0.39.1
 accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0
 tensorrt

setup-runpod.sh

@@ -0,0 +1,14 @@
+# This gets the directory the script is run from so pathing can work relative to the script where needed.
+SCRIPT_DIR="$(cd -- $(dirname -- "$0") && pwd)"
+
+# Check if the venv folder doesn't exist
+if [ ! -d "$SCRIPT_DIR/venv" ]; then
+    echo "Creating venv..."
+    python3 -m venv venv
+fi
+
+# Activate the virtual environment
+source "$SCRIPT_DIR/venv/bin/activate"
+
+# Run setup_linux.py script with platform requirements
+python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_runpod.txt