Spaces:
Running
on
Zero
Running
on
Zero
upgraded the cupy version (#1)
Browse files- upgraded the cupy version (e75aacb4411a285d0987a306c2a4469d912b7f7c)
Co-authored-by: yuvraj sharma <[email protected]>
- 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 |
|