shibing624 commited on
Commit
d06bb8c
·
1 Parent(s): 9608b8c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ pip install -U textgen
37
 
38
  ```python
39
  from textgen import ChatGlmModel
40
- model = ChatGlmModel("chatglm", "THUDM/chatglm-6b", lora_name="shibing624/chatglm-6b-belle-zh-lora")
41
  r = model.predict(["介绍下北京\n答:"])
42
  print(r) # ['北京是中国的首都,是中国的政治、文化和经济中心。...']
43
  ```
 
37
 
38
  ```python
39
  from textgen import ChatGlmModel
40
+ model = ChatGlmModel("chatglm", "THUDM/chatglm-6b", peft_name="shibing624/chatglm-6b-belle-zh-lora")
41
  r = model.predict(["介绍下北京\n答:"])
42
  print(r) # ['北京是中国的首都,是中国的政治、文化和经济中心。...']
43
  ```