Release v0.1.8

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)

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)

requirements.txt

@@ -6,6 +6,7 @@ peft==0.4.0
 trl>=0.7.1
 scipy
 sentencepiece
+protobuf
 tiktoken
 jieba
 rouge-chinese

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"