cunliangkong/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-03-23 10:43:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

modify style

Browse Source 
Former-commit-id: 823af88c3201412da7ef734d34198424e09b2d51
This commit is contained in:
BUAADreamer
2024-05-15 10:18:10 +08:00
parent 0a7e5f2f57
commit 8061cb5671
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/train/sft/trainer.py
View File

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
import numpy as np
import torch
from transformers import Seq2SeqTrainer, ProcessorMixin
from transformers import ProcessorMixin, Seq2SeqTrainer
from ...extras.constants import IGNORE_INDEX
from ...extras.logging import get_logger
@@ -127,4 +127,4 @@ class CustomSeq2SeqTrainer(Seq2SeqTrainer):
if self.processor is not None:
if output_dir is None:
output_dir = self.args.output_dir
getattr(self.processor, "image_processor").save_pretrained(output_dir)
getattr(self.processor, "image_processor").save_pretrained(output_dir)