From 50e7472ae6bc9156684f6e44fad83fecd7428b52 Mon Sep 17 00:00:00 2001 From: theblackcat102 Date: Fri, 6 Jan 2023 17:14:19 +0000 Subject: [PATCH] [fix] push fix by linter --- model/supervised_finetuning/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/model/supervised_finetuning/trainer.py b/model/supervised_finetuning/trainer.py index 2e59edcc..a66f2d11 100644 --- a/model/supervised_finetuning/trainer.py +++ b/model/supervised_finetuning/trainer.py @@ -5,7 +5,6 @@ from typing import Any, Dict, List, Optional, Tuple, Union import torch from torch import nn -from torch.utils.data import Dataset from transformers import PreTrainedModel, Trainer, TrainingArguments, get_cosine_schedule_with_warmup from utils import get_dataset, get_loss, get_model, get_tokenizer, read_yamls