MODLI commited on
Commit
afe455b
·
verified ·
1 Parent(s): 0ac9108

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  fastapi==0.104.1
2
  uvicorn==0.24.0
3
  pillow==10.0.1
4
- torch==2.1.0
5
- transformers==4.35.0
6
  colorthief==0.2.1
7
  python-multipart==0.0.6
8
- accelerate>=0.20.0 # ← IMPORTANT pour optimiser la mémoire
 
 
1
  fastapi==0.104.1
2
  uvicorn==0.24.0
3
  pillow==10.0.1
4
+ torch==2.0.1
5
+ transformers==4.30.2
6
  colorthief==0.2.1
7
  python-multipart==0.0.6
8
+ huggingface-hub==0.16.4
9
+ accelerate==0.20.3