diff --git a/README.md b/README.md index 434f96da..ebb9b779 100644 --- a/README.md +++ b/README.md @@ -116,7 +116,7 @@ huggingface-cli login - Python 3.8+ and PyTorch 1.13.1+ - 🤗Transformers, Datasets, Accelerate, PEFT and TRL -- sentencepiece and tiktoken +- sentencepiece, protobuf and tiktoken - jieba, rouge-chinese and nltk (used at evaluation) - gradio and matplotlib (used in web_demo.py) - uvicorn, fastapi and sse-starlette (used in api_demo.py) diff --git a/README_zh.md b/README_zh.md index 28670ef0..243823e1 100644 --- a/README_zh.md +++ b/README_zh.md @@ -116,7 +116,7 @@ huggingface-cli login - Python 3.8+ 和 PyTorch 1.13.1+ - 🤗Transformers, Datasets, Accelerate, PEFT 和 TRL -- sentencepiece 和 tiktoken +- sentencepiece, protobuf 和 tiktoken - jieba, rouge-chinese 和 nltk (用于评估) - gradio 和 matplotlib (用于网页端交互) - uvicorn, fastapi 和 sse-starlette (用于 API) diff --git a/requirements.txt b/requirements.txt index bb132738..92ef9d76 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,6 +6,7 @@ peft==0.4.0 trl>=0.7.1 scipy sentencepiece +protobuf tiktoken jieba rouge-chinese diff --git a/src/llmtuner/__init__.py b/src/llmtuner/__init__.py index 3b5665f8..320be69c 100644 --- a/src/llmtuner/__init__.py +++ b/src/llmtuner/__init__.py @@ -6,4 +6,4 @@ from llmtuner.tuner import export_model, run_exp from llmtuner.webui import create_ui, create_web_demo -__version__ = "0.1.7" +__version__ = "0.1.8"