winglian commited on
Commit
958da70
1 Parent(s): 759e867

fix formatting

Browse files
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