diff --git a/src/llmtuner/extras/packages.py b/src/llmtuner/extras/packages.py index 26df247b..2e60e690 100644 --- a/src/llmtuner/extras/packages.py +++ b/src/llmtuner/extras/packages.py @@ -18,7 +18,7 @@ _flash_attn2_available = is_package_available("flash_attn") and get_package_vers _jieba_available = is_package_available("jieba") _matplotlib_available = is_package_available("matplotlib") _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") _uvicorn_available = is_package_available("uvicorn")