kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modify some style

Browse Source
This commit is contained in:
BUAADreamer committed 2024-04-25 22:40:53 +08:00
1 parent d29f3798f6
commit ece78a6d6a
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
src/llmtuner/data/preprocess.py
+1
View File
@@ -6,6 +6,7 @@ from ..extras.constants import IGNORE_INDEX
from ..extras.logging import get_logger from ..extras.logging import get_logger
from .utils import Role from .utils import Role
if TYPE_CHECKING: if TYPE_CHECKING:
from transformers import Seq2SeqTrainingArguments from transformers import Seq2SeqTrainingArguments
from transformers.tokenization_utils import AutoProcessor, PreTrainedTokenizer from transformers.tokenization_utils import AutoProcessor, PreTrainedTokenizer