fix formatting
Browse files- scripts/finetune.py +0 -1
- src/axolotl/utils/trainer.py +1 -0
scripts/finetune.py
CHANGED
@@ -261,7 +261,6 @@ def train(
|
|
261 |
model.save_pretrained(cfg.output_dir)
|
262 |
return
|
263 |
|
264 |
-
|
265 |
trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
|
266 |
|
267 |
model.config.use_cache = False
|
|
|
261 |
model.save_pretrained(cfg.output_dir)
|
262 |
return
|
263 |
|
|
|
264 |
trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
|
265 |
|
266 |
model.config.use_cache = False
|
src/axolotl/utils/trainer.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
"""Module containing the Trainer class and related functions"""
|
2 |
|
3 |
import importlib
|
|
|
4 |
import math
|
5 |
import os
|
6 |
import sys
|
|
|
1 |
"""Module containing the Trainer class and related functions"""
|
2 |
|
3 |
import importlib
|
4 |
+
import logging
|
5 |
import math
|
6 |
import os
|
7 |
import sys
|