Hugging Face's logo

Spaces:
Cognomen
/
CatCon-Controlnet-WD-1-5-b2
Runtime error

App Files Community
Cognomen commited on
Commit
0c90b18
·
1 Parent(s): 2da962c

bump jax cuda version to 12

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
- jax[cuda11_pip]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python
@@ -8,6 +8,6 @@ flax
8
  pillow
9
  triton
10
  torch
11
- torchvision
12
  git+https://github.com/huggingface/accelerate
13
  xformers==0.0.16
 
1
  -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
2
+ jax[cuda12_local]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python
 
8
  pillow
9
  triton
10
  torch
11
+ torchvision==0.15.1
12
  git+https://github.com/huggingface/accelerate
13
  xformers==0.0.16