Spaces:
Running
on
Zero
Running
on
Zero
upgraded the cupy version
Browse files- 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-
|
| 13 |
av
|
| 14 |
gpustat
|
| 15 |
|
|
|
|
| 9 |
accelerate==0.30.1
|
| 10 |
transformers==4.41.1
|
| 11 |
colorlog
|
| 12 |
+
cupy-cuda120
|
| 13 |
av
|
| 14 |
gpustat
|
| 15 |
|