Cognomen commited on
Commit
265ced0
Β·
1 Parent(s): 0c90b18

switch to torch 2.0.0

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[cuda12_local]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python
@@ -7,7 +7,7 @@ transformers
7
  flax
8
  pillow
9
  triton
10
- torch
11
  torchvision==0.15.1
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_pip]
3
  jaxlib
4
  git+https://github.com/huggingface/diffusers@main
5
  opencv-python
 
7
  flax
8
  pillow
9
  triton
10
+ torch==2.0.0
11
  torchvision==0.15.1
12
  git+https://github.com/huggingface/accelerate
13
  xformers==0.0.16