Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -5,4 +5,4 @@ tiktoken
|
|
| 5 |
torch
|
| 6 |
torchvision
|
| 7 |
tqdm
|
| 8 |
-
cuda-python==
|
|
|
|
| 5 |
torch
|
| 6 |
torchvision
|
| 7 |
tqdm
|
| 8 |
+
cuda-python==11.8
|