scott/test2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pytorch shit

Browse Source
This commit is contained in:
Scott Register
2025-07-27 08:40:59 -07:00
parent 4cc14bc0a9
commit b5eae7b41d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
runpod_setup.sh
View File

@@ -83,6 +83,10 @@ else
cd .. cd ..
fi 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 # Download SAM2 checkpoints
print_status "Downloading SAM2 checkpoints..." print_status "Downloading SAM2 checkpoints..."
cd segment-anything-2/checkpoints cd segment-anything-2/checkpoints