Hugging Face's logo

Spaces:
FoundationVision
/
UniTok
Running on Zero

App Files Community
1
Junfeng5 commited on
Commit
8162177
·
verified ·
1 Parent(s): dbdaa63

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -6
requirements.txt CHANGED
@@ -1,16 +1,16 @@
1
- torch==2.3.0
2
  torchvision
3
  transformers==4.39.2
4
- timm
5
- deepspeed
 
6
  open_clip_torch==2.32.0
7
  typed-argument-parser
8
  spaces
9
  gradio==4.44.1
10
  gradio_client==1.3.0
11
  pillow
12
- accelerate
13
  bitsandbytes
14
  tqdm
15
- numpy==1.26.4
16
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
1
+ torch==2.4.0
2
  torchvision
3
  transformers==4.39.2
4
+ timm==1.0.8
5
+ deepspeed==0.12.6
6
+ numpy==1.26.4
7
  open_clip_torch==2.32.0
8
  typed-argument-parser
9
  spaces
10
  gradio==4.44.1
11
  gradio_client==1.3.0
12
  pillow
13
+ accelerate==0.27.2
14
  bitsandbytes
15
  tqdm
16
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl