fedorajuandy commited on
Commit
52f84d7
·
1 Parent(s): 24a3da7
Files changed (1) hide show
  1. requirements.txt +5 -13
requirements.txt CHANGED
@@ -1,19 +1,11 @@
1
  # Python package requirements
2
  # Use "pip install -r requrements.txt"
3
 
4
- braceexpand
5
  dalle-mini
6
- datasets[streaming]
7
- einops
8
- emoji
9
- flax
10
- ftfy
11
  gradio
12
- isort
13
- jax
14
- opta
15
- pillow
16
- transformers
17
- tqdm
18
- unidecode
19
  wandb
 
1
  # Python package requirements
2
  # Use "pip install -r requrements.txt"
3
 
4
+ flax==0.6.3 chex==0.1.6 orbax==0.1.2 optaxftfy
5
  dalle-mini
6
+ git+https://github.com/patil-suraj/vqgan-jax.git
7
+ jax==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.html
8
+ jaxlib==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.htmleinops
 
 
9
  gradio
10
+ tensorrt
 
 
 
 
 
 
11
  wandb