Spaces:
Build error
Build error
Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -8,3 +8,4 @@ timm
|
|
| 8 |
einops
|
| 9 |
deepspeed
|
| 10 |
flash-attn==2.3.0
|
|
|
|
|
|
| 8 |
einops
|
| 9 |
deepspeed
|
| 10 |
flash-attn==2.3.0
|
| 11 |
+
cuda-12.1
|