如何保证modelscope中同样的输入内容每次返回的结果也相同呢?

"如何保证modelscope中同样的输入内容每次返回的结果也相同呢?

加载模型和分词器

tokenizer = AutoTokenizer.from_pretrained(""Qwen/Qwen-1_8B-Chat"", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(""Qwen/Qwen-1_8B-Chat"", device_map=""auto"",
trust_remote_code=True).eval()

定义输入字符串

input_strings = [""你好呀"", ""你好呀"", ""你好呀"", ""你好呀""]

for input_text in input_strings:

# 与模型交互
response, _ = model.chat(tokenizer, input_text, history=None,
                         system=""""""【任务:信息分类】\n请严格根据提供分类选项对输入内容进行分类,并直接回复类别名称:\n科技\n就业\n财经\n其它\n"""""")
print(response)"

展开
收起
小小爱吃香菜 2024-03-19 16:31:33 58 分享 版权
1 条回答
写回答
取消 提交回答

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

还有其他疑问?
咨询AI助理