Update training_config.json
Browse files- training_config.json +2 -1
training_config.json
CHANGED
|
@@ -8,5 +8,6 @@
|
|
| 8 |
"context_length": 4098,
|
| 9 |
"train_batch_size": 4,
|
| 10 |
"learning_rate": "2e-4",
|
| 11 |
-
"train_steps": 2000
|
|
|
|
| 12 |
}
|
|
|
|
| 8 |
"context_length": 4098,
|
| 9 |
"train_batch_size": 4,
|
| 10 |
"learning_rate": "2e-4",
|
| 11 |
+
"train_steps": 2000,
|
| 12 |
+
"model_type" : "mistral"
|
| 13 |
}
|