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()