Duskfallcrew commited on
Commit
c84e74b
·
verified ·
1 Parent(s): 5fa0056

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -5
requirements.txt CHANGED
@@ -1,9 +1,10 @@
1
- torch==2.1.0+cu121
2
- diffusers[torch]==0.25.0
3
  transformers==4.36.0
4
  einops==0.6.0
5
  open-clip-torch==2.20.0
6
  invisible-watermark
7
- jax[cuda12_pip]==0.4.23 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
8
- xformers==0.0.23
9
- gradio
 
 
1
+ torch==2.1.0
2
+ diffusers==0.25.0
3
  transformers==4.36.0
4
  einops==0.6.0
5
  open-clip-torch==2.20.0
6
  invisible-watermark
7
+ jax==0.4.23
8
+ gradio
9
+ safetensors
10
+ accelerate