DietNerf-Demo / requirements.txt
alexlau's picture
downgrade jaxlib
0c4c744
raw
history blame
324 Bytes
#--find-links https://storage.googleapis.com/jax-releases/jax_releases.html
#jax[cuda111]
numpy>=1.16.4
jax[cpu]==0.2.17
jaxlib==0.1.68
flax==0.3.4
# opencv-python>=4.4.0
Pillow>=7.2.0
pyyaml>=5.3.1
tensorboard>=2.4.0
tensorflow>=2.3.1
tensorflow-hub>=0.11.0
transformers==4.8.2
tqdm==4.61.2
streamlit==0.84.1
gdown==3.13.0