ModelScope魔搭模型如何搭建webapi呢?

ModelScope魔搭模型如何搭建webapi呢?用flask调用模型出错,而直接运行就不会

展开
收起
真的很搞笑 2023-09-12 19:49:29 152 分享 版权
1 条回答
写回答
取消 提交回答
  • lask理论上也不会,之前用百川的模型写过一个case,import torch
    from flask import Flask, request
    from modelscope import snapshot_download, Model

    app = Flask(name)

    model_dir = snapshot_download("baichuan-inc/Baichuan-13B-Chat", revision='v1.0.3')
    model = Model.from_pretrained(model_dir, device_map="balanced", trust_remote_code=True, torch_dtype=torch.float16)

    @app.route("/")
    def hello_world():
    return "

    Hello, world!

    "

    @app.route("/ask", methods=['POST','GET'])
    def inference():
    question = request.args["question"]
    messages = []
    messages.append({"role": "user", "content": question})
    response = model(messages)
    print(response)
    return "

    %s

    " % response

    app.run("0.0.0.0", "8080", True)供参考,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-09-13 07:48:53
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

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