valegro commited on
Commit
f8a9d39
·
verified ·
1 Parent(s): 14d3133

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch>=2.3 # include CUDA in build GPU
2
  transformers>=4.41
3
- segment-anything==1.0.2
4
- timm
5
  opencv-python-headless
6
  pillow
7
  numpy
 
1
+ torch>=2.3 # la 2.3.0 cu118 è perfetta per la GPU T4
2
  transformers>=4.41
3
+ segment-anything==1.0 # unica release PyPI
4
+ timm # dipendenza SAM
5
  opencv-python-headless
6
  pillow
7
  numpy