Skip to content

Commit

Permalink
comment out torch
Browse files Browse the repository at this point in the history
  • Loading branch information
@shamspias
shamspias committed Sep 15, 2024
1 parent 7576f14 commit a32e2d6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requirements.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,8 @@ boto3
sahi

# PyTorch with CUDA 12.6 support for GPU acceleration
torch==2.0.1+cu126 # PyTorch version with CUDA 12.6
torchvision==0.15.2+cu126 # TorchVision with CUDA 12.6
# torch==2.0.1+cu124 # PyTorch version with CUDA 12.6
# torchvision==0.15.2+cu124 # TorchVision with CUDA 12.6

# Additional dependencies for models or utilities
matplotlib # For plotting images (used in sahi_utils.py)

0 comments on commit a32e2d6

Please sign in to comment.