From b5eae7b41d2f0501baeeed9dd2a7ab332dc2fa93 Mon Sep 17 00:00:00 2001
From: Scott Register <sreg@sreg.io>
Date: Sun, 27 Jul 2025 08:40:59 -0700
Subject: [PATCH] pytorch shit

---
 runpod_setup.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/runpod_setup.sh b/runpod_setup.sh
index 34b635d..1d6f301 100755
--- a/runpod_setup.sh
+++ b/runpod_setup.sh
@@ -83,6 +83,10 @@ else
     cd ..
 fi
 
+# Fix PyTorch version conflicts after SAM2 installation
+print_status "Fixing PyTorch version conflicts..."
+pip install torchaudio --upgrade --no-deps || print_error "Failed to upgrade torchaudio"
+
 # Download SAM2 checkpoints
 print_status "Downloading SAM2 checkpoints..."
 cd segment-anything-2/checkpoints