wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-06-27 16:10:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

[fix] transformers import error

Browse Source
This commit is contained in:
theblackcat102
2023-02-07 01:26:28 +00:00
parent af1c62cd83
commit a39cbab524
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/supervised_finetuning/trainer.py
+2 -1
View File
@@ -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