Spaces:
Running
Running
cuda-python==12.6.2.post1
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -16,4 +16,5 @@ flash-attn==1.0.5
|
|
| 16 |
torch==2.5.1
|
| 17 |
nvidia-cuda-nvcc==0.0.1.dev5
|
| 18 |
nvidia-cuda-nvcc-cu11==11.8.89
|
|
|
|
| 19 |
torchvision==0.20.1
|
|
|
|
| 16 |
torch==2.5.1
|
| 17 |
nvidia-cuda-nvcc==0.0.1.dev5
|
| 18 |
nvidia-cuda-nvcc-cu11==11.8.89
|
| 19 |
+
cuda-python==12.6.2.post1
|
| 20 |
torchvision==0.20.1
|