pytorch shit

2025-07-27 08:40:59 -07:00
parent 4cc14bc0a9
commit b5eae7b41d
1 changed files with 4 additions and 0 deletions
--- 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