diff --git a/README.md b/README.md index ef434ac1..19af693c 100644 --- a/README.md +++ b/README.md @@ -125,7 +125,7 @@ huggingface-cli login - Python 3.8+ and PyTorch 1.13.1+ - 🤗Transformers, Datasets, Accelerate, PEFT and TRL - sentencepiece, protobuf and tiktoken -- jieba, rouge-chinese and nltk (used at evaluation) +- fire, jieba, rouge-chinese and nltk (used at evaluation and predict) - gradio and matplotlib (used in web_demo.py) - uvicorn, fastapi and sse-starlette (used in api_demo.py) @@ -478,7 +478,7 @@ If this work is helpful, please kindly cite as: ## Acknowledgement -This repo benefits from [PEFT](https://github.com/huggingface/peft), [QLoRA](https://github.com/artidoro/qlora), [FastChat](https://github.com/lm-sys/FastChat) and [OpenChatKit](https://github.com/togethercomputer/OpenChatKit). Thanks for their wonderful works. +This repo benefits from [PEFT](https://github.com/huggingface/peft), [QLoRA](https://github.com/artidoro/qlora) and [FastChat](https://github.com/lm-sys/FastChat). Thanks for their wonderful works. ## Star History diff --git a/README_zh.md b/README_zh.md index 35f72c47..386536a1 100644 --- a/README_zh.md +++ b/README_zh.md @@ -125,7 +125,7 @@ huggingface-cli login - Python 3.8+ 和 PyTorch 1.13.1+ - 🤗Transformers, Datasets, Accelerate, PEFT 和 TRL - sentencepiece, protobuf 和 tiktoken -- jieba, rouge-chinese 和 nltk (用于评估) +- fire, jieba, rouge-chinese 和 nltk (用于评估及预测) - gradio 和 matplotlib (用于网页端交互) - uvicorn, fastapi 和 sse-starlette (用于 API) @@ -477,7 +477,7 @@ CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \ ## 致谢 -本项目受益于 [PEFT](https://github.com/huggingface/peft)、[QLoRA](https://github.com/artidoro/qlora)、[FastChat](https://github.com/lm-sys/FastChat) 和 [OpenChatKit](https://github.com/togethercomputer/OpenChatKit),感谢以上诸位作者的付出。 +本项目受益于 [PEFT](https://github.com/huggingface/peft)、[QLoRA](https://github.com/artidoro/qlora) 和 [FastChat](https://github.com/lm-sys/FastChat),感谢以上诸位作者的付出。 ## Star History diff --git a/requirements.txt b/requirements.txt index 2eb25a2c..c132683b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,6 +8,7 @@ scipy sentencepiece protobuf tiktoken +fire jieba rouge-chinese nltk