gron371 commited on
Commit
b5f6c99
·
verified ·
1 Parent(s): b15e27e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,12 +1,12 @@
 
 
 
 
1
  fastapi
2
  python-multipart
3
  uvicorn
4
  Pillow==10.1.0
5
- torch==2.1.2
6
- torchvision==0.16.2
7
  transformers==4.40.0
8
  sentencepiece==0.1.99
9
  decord
10
- huggingface_hub
11
- numpy<2.0.0
12
- flash-attn
 
1
+ numpy<2.0.0
2
+ torch==2.1.2
3
+ torchvision==0.16.2
4
+ --extra-index-url https://download.pytorch.org/whl/cu118
5
  fastapi
6
  python-multipart
7
  uvicorn
8
  Pillow==10.1.0
 
 
9
  transformers==4.40.0
10
  sentencepiece==0.1.99
11
  decord
12
+ huggingface_hub