train model
Browse files
scripts/requirements-lit.in
CHANGED
@@ -5,6 +5,7 @@ jinja2
|
|
5 |
transformers
|
6 |
bitsandbytes
|
7 |
wandb
|
8 |
-
litgpt[all]
|
|
|
9 |
litdata
|
10 |
grokadamw
|
|
|
5 |
transformers
|
6 |
bitsandbytes
|
7 |
wandb
|
8 |
+
# litgpt[all]
|
9 |
+
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
10 |
litdata
|
11 |
grokadamw
|