Hugging Face's logo

Spaces:
OneEyeDJ
/
test
Sleeping

App Files Community
OneEyeDJ commited on
Commit
9d7d1a9
·
verified ·
1 Parent(s): 20d3244

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,12 +1,12 @@
1
  ###############################################################################
2
- # CUDA 12.2 + PyTorch 2.1 (ZeroGPU-friendly)
3
  ###############################################################################
4
  --extra-index-url https://download.pytorch.org/whl/cu122
5
  --upgrade
6
  --force-reinstall
7
 
8
- torch==2.1.0 # plain spec → ZeroGPU accepts it
9
- torchvision==0.16.0
10
 
11
  ###############################################################################
12
  # Transformers / training utils
 
1
  ###############################################################################
2
+ # CUDA 12.2 + PyTorch 2.1 (overwrite CPU wheel from the base image)
3
  ###############################################################################
4
  --extra-index-url https://download.pytorch.org/whl/cu122
5
  --upgrade
6
  --force-reinstall
7
 
8
+ torch==2.1.0+cu122
9
+ torchvision==0.16.0+cu122
10
 
11
  ###############################################################################
12
  # Transformers / training utils