pretrain core 0
Browse files- scripts/requirements.in +1 -1
scripts/requirements.in
CHANGED
|
@@ -10,7 +10,7 @@ transformers
|
|
| 10 |
wandb
|
| 11 |
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
| 12 |
mergekit @ git+https://github.com/arcee-ai/mergekit.git
|
| 13 |
-
|
| 14 |
# bitsandbytes
|
| 15 |
# grokadamw
|
| 16 |
# sophia-opt
|
|
|
|
| 10 |
wandb
|
| 11 |
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 |
# grokadamw
|
| 16 |
# sophia-opt
|