erfanzar commited on
Commit
6df7386
·
1 Parent(s): a822df8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  jax~=0.4.13
 
2
  flax~=0.7.1
3
  fjutils~=0.0.15
4
  numpy==1.25.2
@@ -12,4 +13,5 @@ tqdm==4.65.0
12
  datasets==2.14.3
13
  setuptools~=60.0.0
14
  easydel
15
- wandb
 
 
1
  jax~=0.4.13
2
+ jaxlib~=0.4.13
3
  flax~=0.7.1
4
  fjutils~=0.0.15
5
  numpy==1.25.2
 
13
  datasets==2.14.3
14
  setuptools~=60.0.0
15
  easydel
16
+ wandb
17
+ torch