From f901963bafdea7740c1aa92c0e4e9b931fdbd0a8 Mon Sep 17 00:00:00 2001 From: Cherrytest Date: Sat, 21 Oct 2023 00:22:41 +0000 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d3439ee..f2ec253 100644 --- a/README.md +++ b/README.md @@ -28,8 +28,8 @@ import torch from modelscope import Model, AutoTokenizer -model = Model.from_pretrained("ZhipuAI/agentlm-7b", revision='master', device_map='auto', torch_dtype=torch.float16) -tokenizer = AutoTokenizer.from_pretrained("ZhipuAI/agentlm-7b", revision='master') +model = Model.from_pretrained("ZhipuAI/agentlm-13b", revision='master', device_map='auto', torch_dtype=torch.float16) +tokenizer = AutoTokenizer.from_pretrained("ZhipuAI/agentlm-13b", revision='master') prompt = """[INST] <> [INST] <>