myniu commited on
Commit
4837b64
1 Parent(s): f925ba2
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -9,7 +9,7 @@ einops
9
  accelerate==0.30.1
10
  transformers==4.41.1
11
  colorlog
12
- cupy-cuda120
13
  av
14
  gpustat
15
 
 
9
  accelerate==0.30.1
10
  transformers==4.41.1
11
  colorlog
12
+ cupy-cuda12x
13
  av
14
  gpustat
15