From a39cbab524d38a9891153528285881a6923e3eb7 Mon Sep 17 00:00:00 2001 From: theblackcat102 Date: Tue, 7 Feb 2023 01:26:28 +0000 Subject: [PATCH] [fix] transformers import error --- model/supervised_finetuning/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model/supervised_finetuning/trainer.py b/model/supervised_finetuning/trainer.py index 043534ea..c500f8df 100644 --- a/model/supervised_finetuning/trainer.py +++ b/model/supervised_finetuning/trainer.py @@ -9,7 +9,8 @@ import torch from torch import nn from torch.utils.data import DataLoader from transformers import PreTrainedModel, Trainer, TrainingArguments -from transformers.trainer_pt_utils import IterableDatasetShard, seed_worker +from transformers.trainer_pt_utils import IterableDatasetShard +from transformers.trainer_utils import seed_worker from transformers.training_args import OptimizerNames from transformers.utils import is_datasets_available from utils import get_dataset, get_loss, get_metrics, get_model, get_tokenizer, read_yamls