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
|