fix formatting

This commit is contained in:
Wing Lian
2023-06-10 15:28:08 -04:00
parent 759e8673ce
commit 958da70376
2 changed files with 1 additions and 1 deletions

View File

@@ -261,7 +261,6 @@ def train(
model.save_pretrained(cfg.output_dir)
return
trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
model.config.use_cache = False

View File

@@ -1,6 +1,7 @@
"""Module containing the Trainer class and related functions"""
import importlib
import logging
import math
import os
import sys