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] <>