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 53 分享 版权
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助理