Update README.md
Browse files
README.md
CHANGED
|
@@ -20,5 +20,5 @@ tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote
|
|
| 20 |
model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True).half().cuda()
|
| 21 |
model = model.eval()
|
| 22 |
response, history = model.chat(tokenizer, "你好", history=[])
|
| 23 |
-
|
| 24 |
```
|
|
|
|
| 20 |
model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True).half().cuda()
|
| 21 |
model = model.eval()
|
| 22 |
response, history = model.chat(tokenizer, "你好", history=[])
|
| 23 |
+
response
|
| 24 |
```
|