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
|