gigant commited on
Commit
a2ae2ef
1 Parent(s): d906598

Update requirements.txt

Browse files

jax tpu requirement is not needed on the space, but gpu is

Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  git+https://github.com/huggingface/diffusers.git #diffusers==0.16.0.dev0
2
  -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
3
- jax[tpu]==0.4.5
4
  absl-py==1.4.0
5
  accelerate==0.16.0
6
  addict==2.4.0
@@ -29,7 +29,6 @@ cycler==0.11.0
29
  datasets==2.11.0
30
  decorator==4.4.2
31
  decord==0.6.0
32
- # diffusers==0.16.0.dev0
33
  dill==0.3.6
34
  dm-tree==0.1.8
35
  einops==0.6.0
 
1
  git+https://github.com/huggingface/diffusers.git #diffusers==0.16.0.dev0
2
  -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
3
+ jax[gpu]#jax[tpu]==0.4.5
4
  absl-py==1.4.0
5
  accelerate==0.16.0
6
  addict==2.4.0
 
29
  datasets==2.11.0
30
  decorator==4.4.2
31
  decord==0.6.0
 
32
  dill==0.3.6
33
  dm-tree==0.1.8
34
  einops==0.6.0