训练文本生成

18 lines
452 B

2 years ago
from .common import (
load_pretrained,
prepare_args,
prepare_infer_args,
2 years ago
prepare_data,
preprocess_data
)
from .data_collator import DynamicDataCollatorWithPadding
2 years ago
from .peft_trainer import PeftTrainer, LogCallback
from .seq2seq import ComputeMetrics, Seq2SeqPeftTrainer
from .pairwise import PairwiseDataCollatorWithPadding, PairwisePeftTrainer
from .ppo import PPOPeftTrainer
2 years ago
from .other import get_logits_processor, plot_loss