From ccb3553576164113c31be714a0295ea82321d67d Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Mon, 11 Sep 2023 17:31:34 +0800
Subject: [PATCH] Release v0.1.8

---
 README.md                | 2 +-
 README_zh.md             | 2 +-
 requirements.txt         | 1 +
 src/llmtuner/__init__.py | 2 +-
 4 files changed, 4 insertions(+), 3 deletions(-)

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"