Yaron Koresh commited on
Commit
34044da
·
verified ·
1 Parent(s): 83ea3df

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -11,6 +11,7 @@ transformers
11
  pathos
12
  dill
13
  jax
 
14
  --extra-index-url https://download.pytorch.org/whl/cu121
15
  torch
16
  torchaudio
 
11
  pathos
12
  dill
13
  jax
14
+ flax
15
  --extra-index-url https://download.pytorch.org/whl/cu121
16
  torch
17
  torchaudio