423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 16:18:10 +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: a78b11d944b6cb7dbe2a1d8a24d240e196aa530a
This commit is contained in:
Shaowen Wang 2023-11-16 20:12:35 -06:00 committed by GitHub
parent 77cb18e9e3
commit 07f934566a
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")