Yaron Koresh commited on
Commit
616aab0
·
verified ·
1 Parent(s): f5d0e22

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -4
requirements.txt CHANGED
@@ -7,11 +7,15 @@ invisible_watermark
7
  huggingface_hub
8
  beautifulsoup4
9
  bs4
10
- peft >= 0.6.0
11
- transformers >= 4.34.0
12
- pathos
13
- dill
14
  flax
 
 
 
 
 
 
15
  --extra-index-url https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
16
  jax[cuda_local]
17
  jaxlib[cuda_local]
 
7
  huggingface_hub
8
  beautifulsoup4
9
  bs4
10
+ peft>=0.6.0
11
+ transformers>=4.34.0
 
 
12
  flax
13
+ cython>=0.29.30
14
+ numpy>=1.0
15
+ mpi4py>=1.3
16
+ dill>=0.3.8
17
+ pox>=0.3.4
18
+ pathos>=0.3.2
19
  --extra-index-url https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
20
  jax[cuda_local]
21
  jaxlib[cuda_local]