Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
@@ -2,3 +2,4 @@ torch
|
|
2 |
Pillow
|
3 |
torchvision
|
4 |
timm
|
|
|
|
2 |
Pillow
|
3 |
torchvision
|
4 |
timm
|
5 |
+
cupy
|