开发者社区 > ModelScope模型即服务 > 正文

ModelScope的pipeline 的taskName 应该写什么?

from modelscope.models import Model
model = Model.from_pretrained('qwen/Qwen2-0.5B')
from modelscope.pipelines import pipeline
open_domain_conversation = pipeline('open-domain-conversation', model=model, device='cpu')
print(open_domain_conversation("什么是地球"))

ModelScope的pipeline 的taskName 应该写什么?

展开
收起
小小爱吃香菜 2024-07-23 22:02:10 24 0
1 条回答
写回答
取消 提交回答
  • 北京阿里云ACE会长

    对于 pipeline 函数,taskName 参数应该指定你想要执行的任务类型。在你的例子中,taskName 应该是 'open-domain-conversation',这表示你想要使用这个管道来进行开放域的对话任务。

    from modelscope.models import Model
    model = Model.from_pretrained('qwen/Qwen2-0.5B')
    from modelscope.pipelines import pipeline
    
    # 使用 'open-domain-conversation' 作为 taskName
    open_domain_conversation = pipeline('open-domain-conversation', model=model, device='cpu')
    print(open_domain_conversation("什么是地球"))
    
    2024-07-24 10:05:08
    赞同 1 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载