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

Fix: Change rouge-chinese package name to rouge_chinese

Browse Source 
To reproduce:
python:
importlib.util.find_spec('rouge-chinese') -> None
importlib.util.find_spec('rouge_chinese') -> ModuleSpec(name='rouge_chinese'...)
from rouge_chinese import Rouge
print(Rouge.__module__) -> rouge_chinese
Former-commit-id: 397e9489849d80517b0c467852a7092a0a9626f2
This commit is contained in:
Shaowen Wang 2023-11-16 20:12:35 -06:00 committed by GitHub
parent 5de45bf989
commit 4ea3144554
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/extras/packages.py
View File

@ -18,7 +18,7 @@ _flash_attn2_available = is_package_available("flash_attn") and get_package_vers
_jieba_available = is_package_available("jieba") _jieba_available = is_package_available("jieba")
_matplotlib_available = is_package_available("matplotlib") _matplotlib_available = is_package_available("matplotlib")
_nltk_available = is_package_available("nltk") _nltk_available = is_package_available("nltk")
_rouge_available = is_package_available("rouge-chinese") _rouge_available = is_package_available("rouge_chinese")
_starlette_available = is_package_available("sse-starlette") _starlette_available = is_package_available("sse-starlette")
_uvicorn_available = is_package_available("uvicorn") _uvicorn_available = is_package_available("uvicorn")