Hugging Face's logo

Spaces:
FoundationVision
/
UniTok
Running on Zero

App Files Community
1
Junfeng5 commited on
Commit
ad8046a
·
verified ·
1 Parent(s): 47fe8ba

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,8 +1,10 @@
1
- torch==2.4.0
2
  torchvision
3
  transformers==4.39.2
4
  timm
5
  deepspeed
 
 
6
  spaces
7
  gradio==4.44.1
8
  gradio_client==1.3.0
@@ -10,4 +12,4 @@ pillow
10
  accelerate
11
  bitsandbytes
12
  tqdm
13
- 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
 
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
 
12
  accelerate
13
  bitsandbytes
14
  tqdm
15
+ 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