From 21d8c78edfde1d829be33168138c947d7c60fefe Mon Sep 17 00:00:00 2001 From: Aohan Zeng Date: Tue, 14 Mar 2023 02:48:06 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 1e02694..3152638 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ ChatGLM-6B 是一个开源的、支持中英双语问答的对话语言模型, 可以通过如下代码调用 ChatGLM-6B 模型来生成对话: -```ipython +```python >>> from transformers import AutoTokenizer, AutoModel >>> tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True) >>> model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()