modify some style

Former-commit-id: 1291c7ee39361dd75247c67f04dcf20b472faf83
This commit is contained in:
BUAADreamer 2024-04-25 22:40:53 +08:00
parent 8506eeac13
commit 7e369fd58e

View File

@ -6,6 +6,7 @@ from ..extras.constants import IGNORE_INDEX
from ..extras.logging import get_logger
from .utils import Role
if TYPE_CHECKING:
from transformers import Seq2SeqTrainingArguments
from transformers.tokenization_utils import AutoProcessor, PreTrainedTokenizer