Ryanfafa commited on
Commit
d385ed9
·
verified ·
1 Parent(s): e334e70

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  torch==2.2.2+cpu
2
  torchvision==0.17.2+cpu
3
  --extra-index-url https://download.pytorch.org/whl/cpu
@@ -5,8 +6,10 @@ torchvision==0.17.2+cpu
5
  fastapi
6
  uvicorn[standard]
7
  pillow
8
- transformers>=5.0.0
9
- numpy
 
 
10
  tqdm
11
  tensorboard
12
  nltk
 
1
+ # Torch is installed in the Dockerfile, keep versions consistent
2
  torch==2.2.2+cpu
3
  torchvision==0.17.2+cpu
4
  --extra-index-url https://download.pytorch.org/whl/cpu
 
6
  fastapi
7
  uvicorn[standard]
8
  pillow
9
+ # Use a transformers version compatible with torch 2.2
10
+ transformers==4.46.3
11
+ # Avoid NumPy 2.x issues
12
+ numpy<2.0
13
  tqdm
14
  tensorboard
15
  nltk