Malaji71 commited on
Commit
6875677
·
verified ·
1 Parent(s): 482876d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- # Core ML frameworks - CPU optimized versions
2
- torch==2.1.0+cpu
3
- torchvision==0.16.0+cpu
4
 
5
  # Transformers for Florence-2 model
6
  transformers==4.44.0
 
1
+ # Core ML frameworks - Standard versions for ZeroGPU compatibility
2
+ torch==2.1.0
3
+ torchvision==0.16.0
4
 
5
  # Transformers for Florence-2 model
6
  transformers==4.44.0