pretrain model, lightning-thunder
Browse files- scripts/requirements.in +5 -1
scripts/requirements.in
CHANGED
|
@@ -11,4 +11,8 @@ litdata
|
|
| 11 |
grokadamw
|
| 12 |
# bitsandbytes
|
| 13 |
# pyzstd
|
| 14 |
-
zstd
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
grokadamw
|
| 12 |
# bitsandbytes
|
| 13 |
# pyzstd
|
| 14 |
+
# zstd
|
| 15 |
+
|
| 16 |
+
nvfuser-cu121-torch24
|
| 17 |
+
nvidia-cudnn-frontend
|
| 18 |
+
lightning-thunder
|