ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clean code

Browse Source 
Former-commit-id: f54cafd5c7f0383370d1a2f357834a61a97397ce
This commit is contained in:
hiyouga
2024-06-13 01:58:16 +08:00
parent 04d7629abf
commit 0a75224f62
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: