marcelbinz commited on
Commit
c7bed40
·
verified ·
1 Parent(s): 298c3b7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -5,7 +5,8 @@ torch==2.5.1+cu124 # keep before flash-a
5
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.0.post2/flash_attn-2.8.0.post2+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl # <- 240 MB wheel:contentReference[oaicite:2]{index=2}
6
 
7
  transformers>=4.52.0
8
- accelerate>=0.30.2 # bug-fix for device_map edge case:contentReference[oaicite:3]{index=3}
9
  bitsandbytes==0.43.3
10
- gradio>=4.44.0 # Zero-GPU queue fix PR #5698:contentReference[oaicite:4]{index=4}
 
11
  sentencepiece
 
5
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.0.post2/flash_attn-2.8.0.post2+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl # <- 240 MB wheel:contentReference[oaicite:2]{index=2}
6
 
7
  transformers>=4.52.0
8
+ accelerate>=0.30.2
9
  bitsandbytes==0.43.3
10
+ peft==0.15.2
11
+ gradio>=4.44.0
12
  sentencepiece