text-to-image-generation / requirements.txt
fedorajuandy's picture
add to remove ImportError: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html
9b3a492
raw
history blame contribute delete
369 Bytes
# Python package requirements
# Use "pip install -r requrements.txt"
flax==0.6.3
chex==0.1.6
orbax==0.1.2
optax
dalle-mini
git+https://github.com/patil-suraj/vqgan-jax.git
jax==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.html
jaxlib==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.htmleinops
gradio
tensorrt
wandb
ipywidgets