myniu commited on
Commit
7dc1261
1 Parent(s): 8b12e95
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-cuda117
13
  av
14
  gpustat
15
 
 
9
  accelerate==0.30.1
10
  transformers==4.41.1
11
  colorlog
12
+ cupy-cuda113
13
  av
14
  gpustat
15