Update README.md
This commit is contained in:
@@ -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 = """<s>[INST] <<SYS>>
|
||||
<s>[INST] <<SYS>>
|
||||
|
||||
Reference in New Issue
Block a user