Update README.md

This commit is contained in:
Cherrytest
2023-10-21 00:22:41 +00:00
parent b94edc955e
commit f901963baf

View File

@@ -28,8 +28,8 @@ import torch
from modelscope import Model, AutoTokenizer from modelscope import Model, AutoTokenizer
model = Model.from_pretrained("ZhipuAI/agentlm-7b", revision='master', device_map='auto', torch_dtype=torch.float16) model = Model.from_pretrained("ZhipuAI/agentlm-13b", revision='master', device_map='auto', torch_dtype=torch.float16)
tokenizer = AutoTokenizer.from_pretrained("ZhipuAI/agentlm-7b", revision='master') tokenizer = AutoTokenizer.from_pretrained("ZhipuAI/agentlm-13b", revision='master')
prompt = """<s>[INST] <<SYS>> prompt = """<s>[INST] <<SYS>>
<s>[INST] <<SYS>> <s>[INST] <<SYS>>