pretrain
Browse files- scripts/requirements.in +2 -2
scripts/requirements.in
CHANGED
@@ -12,8 +12,8 @@ litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
|
12 |
mergekit @ git+https://github.com/arcee-ai/mergekit.git
|
13 |
torchao
|
14 |
# bitsandbytes
|
15 |
-
|
16 |
-
sophia-opt
|
17 |
# dolphinflow
|
18 |
# unsloth
|
19 |
lm_eval[ifeval,math]
|
|
|
12 |
mergekit @ git+https://github.com/arcee-ai/mergekit.git
|
13 |
torchao
|
14 |
# bitsandbytes
|
15 |
+
grokadamw
|
16 |
+
# sophia-opt
|
17 |
# dolphinflow
|
18 |
# unsloth
|
19 |
lm_eval[ifeval,math]
|