THUdyh commited on
Commit
bd0ffd1
1 Parent(s): 69a4d07

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  huggingface_hub==0.22.2
2
  torch==2.1.2
3
- flash_attn==2.5.6
4
  torchvision
5
  transformers==4.39.2
6
  tokenizers==0.15.2
@@ -11,7 +11,6 @@ peft==0.4.0
11
  bitsandbytes==0.41.0
12
  pydantic<2,>=1
13
  markdown2
14
- numpy<2
15
  scikit-learn==1.2.2
16
  gradio==3.35.2
17
  gradio_client==0.2.9
@@ -25,4 +24,5 @@ timm==0.9.16
25
  decord
26
  ninja
27
  deepspeed==0.12.2
28
- protobuf
 
 
1
  huggingface_hub==0.22.2
2
  torch==2.1.2
3
+ numpy==1.26.4
4
  torchvision
5
  transformers==4.39.2
6
  tokenizers==0.15.2
 
11
  bitsandbytes==0.41.0
12
  pydantic<2,>=1
13
  markdown2
 
14
  scikit-learn==1.2.2
15
  gradio==3.35.2
16
  gradio_client==0.2.9
 
24
  decord
25
  ninja
26
  deepspeed==0.12.2
27
+ protobuf
28
+ flash_attn==2.5.6