Update README.md
This commit is contained in:
@@ -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>>
|
||||||
|
|||||||
Reference in New Issue
Block a user