prithivMLmods commited on
Commit
6557856
·
verified ·
1 Parent(s): 5e5fd3d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -7
requirements.txt CHANGED
@@ -1,9 +1,18 @@
1
- gradio==6.3.0
 
 
 
 
 
2
  huggingface_hub
3
- hf_xet
4
- torch
5
  torchvision
6
- transformers==4.57.1
7
- bitsandbytes
8
- git+https://github.com/vllm-project/llm-compressor.git
9
- flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
 
 
 
 
 
1
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
2
+ git+https://github.com/huggingface/transformers.git@v4.46.3
3
+ git+https://github.com/huggingface/accelerate.git
4
+ git+https://github.com/huggingface/diffusers.git
5
+ git+https://github.com/huggingface/peft.git
6
+ tokenizers==0.20.3
7
  huggingface_hub
8
+ sentencepiece
9
+ torch==2.6.0
10
  torchvision
11
+ matplotlib
12
+ accelerate
13
+ easydict
14
+ addict
15
+ gradio # - gradio@6.3.0
16
+ einops
17
+ spaces
18
+ numpy