Spaces:

THUdyh
/

Oryx

Running on Zero

THUdyh commited on Sep 25

Commit

bd0ffd1

•

1 Parent(s): 69a4d07

Update requirements.txt

Files changed (1) hide show

requirements.txt CHANGED Viewed

@@ -1,6 +1,6 @@
 huggingface_hub==0.22.2
 torch==2.1.2
-flash_attn==2.5.6
 torchvision
 transformers==4.39.2
 tokenizers==0.15.2
@@ -11,7 +11,6 @@ peft==0.4.0
 bitsandbytes==0.41.0
 pydantic<2,>=1
 markdown2
-numpy<2
 scikit-learn==1.2.2
 gradio==3.35.2
 gradio_client==0.2.9
@@ -25,4 +24,5 @@ timm==0.9.16
 decord
 ninja
 deepspeed==0.12.2
-protobuf

 huggingface_hub==0.22.2
 torch==2.1.2
+numpy==1.26.4
 torchvision
 transformers==4.39.2
 tokenizers==0.15.2
 bitsandbytes==0.41.0
 pydantic<2,>=1
 markdown2
 scikit-learn==1.2.2
 gradio==3.35.2
 gradio_client==0.2.9
 decord
 ninja
 deepspeed==0.12.2
+protobuf
+flash_attn==2.5.6