423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 20:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean code

Browse Source
This commit is contained in:
hiyouga
2024-06-13 01:58:16 +08:00
parent 1f23f25226
commit 2ed8270112
4 changed files with 17 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/train/sft/metric.py
View File

@@ -2,9 +2,10 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING, Dict, Sequence, Tuple, Union
import numpy as np
from transformers.utils import is_jieba_available, is_nltk_available
from ...extras.constants import IGNORE_INDEX
from ...extras.packages import is_jieba_available, is_nltk_available, is_rouge_available
from ...extras.packages import is_rouge_available
if TYPE_CHECKING: