Update README.md

README.md

@@ -12,7 +12,7 @@ widget:
 
 ---
 
-# Chinese Spelling Correction LoRA Model
+# Chinese QA LoRA Model
 ChatGLM中文问答LoRA模型
 
 `chatglm-6b-belle-zh-lora` evaluate test data：
@@ -23,27 +23,27 @@ The overall performance of chatglm-6b-belle-zh-lora on QA **test**:
 |:-- |:--- |
 |介绍下北京|北京是中国的首都，是中国的政治、文化和经济中心。它位于中国的北部，拥有悠久的历史和文化遗产，如故宫、长城和天坛等。北京也是中国最著名的城市之一，有着丰富的美食、文化和娱乐活动。北京还有许多旅游景点，如颐和园、圆明园和天坛等。北京是一个充满活力和机会的城市，吸引了来自全国和世界各地的人们前来探索。|
 
-在中文开放测试集中的表现优异，继承了两方面的优势：1）微调的底座是ChatGLM-6B模型，中文的表现优于LLAMA，2）微调使用的100万高质量的中文Belle数据集，微调后的模型对话效果优于原始ChatGLM-6B。
+在中文开放测试集中的表现优异，继承了两方面的优势：1）微调的底座是ChatGLM-6B模型，中文的表现优于LLAMA，2）微调使用的是高质量100万条中文ChatGPT指令Belle数据集，微调后的模型对话效果优于原始ChatGLM-6B。
 
 
 ## Usage
 
-本项目开源在lmft项目：[lmft](https://github.com/shibing624/lmft)，可支持ChatGLM模型，通过如下命令调用：
+本项目开源在textgen项目：[textgen](https://github.com/shibing624/textgen)，可支持ChatGLM模型，通过如下命令调用：
 
 Install package:
 ```shell
-pip install -U lmft
+pip install -U textgen
 ```
 
 ```python
-from lmft import ChatGlmModel
+from textgen import ChatGlmModel
 model = ChatGlmModel("chatglm", "THUDM/chatglm-6b", lora_name="shibing624/chatglm-6b-belle-zh-lora")
 r = model.predict(["介绍下北京\n答："])
 print(r) # ['北京是中国的首都，是中国的政治、文化和经济中心。...']
 ```
 
 ## Usage (HuggingFace Transformers)
-Without [lmft](https://github.com/shibing624/lmft), you can use the model like this: 
+Without [textgen](https://github.com/shibing624/textgen), you can use the model like this: 
 
 First, you pass your input through the transformer model, then you get the generated sentence.
 
@@ -94,19 +94,16 @@ chatglm-6b-belle-zh-lora
 5. 69万条中文指令Guanaco数据集(Belle50万条+Guanaco19万条)：[Chinese-Vicuna/guanaco_belle_merge_v1.0](https://huggingface.co/datasets/Chinese-Vicuna/guanaco_belle_merge_v1.0)
 
 
-如果需要训练ChatGLM模型，请参考[https://github.com/shibing624/lmft](https://github.com/shibing624/lmft)
+如果需要训练ChatGLM模型，请参考[https://github.com/shibing624/textgen](https://github.com/shibing624/textgen)
 
 
 ## Citation
 
 ```latex
-@software{lmft,
+@software{textgen,
   author = {Xu Ming},
-  title = {lmft: Implementation of language model finetune},
-  year = {2023},
-  url = {https://github.com/shibing624/lmft},
+  title = {textgen: Implementation of language model finetune},
+  year = {2021},
+  url = {https://github.com/shibing624/textgen},
 }
 ```
-
-
-