Commit
·
d06bb8c
1
Parent(s):
9608b8c
Update README.md
Browse files
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",
|
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 |
```
|