Junfeng5 commited on
Commit
61ec284
·
verified ·
1 Parent(s): ad8046a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -12,4 +12,5 @@ pillow
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
 
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