Lev Israel Claude Opus 4.6 commited on
Commit
d1268eb
·
1 Parent(s): b044089

Pin torch<2.7 for CUDA 12.0 compat on HF Spaces A100

Browse files

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -2,8 +2,8 @@
2
  gradio>=5.0.0
3
  transformers>=4.36.0
4
  sentence-transformers>=2.2.2
5
- torch>=2.0.0
6
- torchvision>=0.15.0
7
  datasets>=2.14.0
8
  huggingface_hub>=0.19.0
9
 
 
2
  gradio>=5.0.0
3
  transformers>=4.36.0
4
  sentence-transformers>=2.2.2
5
+ torch>=2.0.0,<2.7.0
6
+ torchvision>=0.15.0,<0.22.0
7
  datasets>=2.14.0
8
  huggingface_hub>=0.19.0
9