开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

在ModelScope中,推理的时候报了错怎么解决?

帮忙看看这个问题dee5f0c1fc4a96d443d88643d446bfbf.png
huatuo7b推理的时候报了错38d1fe9e36c0c5840ab7e7de9117e78e.png

展开
收起
崔问问 2023-12-03 23:18:14 46 0
2 条回答
写回答
取消 提交回答
  • 根据您提供的代码和错误信息,我发现了一些问题。以下是修改后的代码:

    import torch
    from modelscope.pipelines import pipeline
    from modelscope.utils.constant import Tasks
    
    model_dir = 'zhouk123/Llama-2-7b-ms-huatuo'
    snapshot_download(model_id=model_dir, cache_dir='huatuo2', revision='master')
    
    tokenizer = AutoTokenizer.from_pretrained(model_dir, device_map="auto", trust_remote_code=True)
    model = AutoModel.from_pretrained(model_dir, device_map="auto")
    model.generation_config = GenerationConfig.from_pretrained(model_dir)
    trust_remote_code=True, torch_dtype=torch.float16)
    
    messages = []
    messages.append({"role": "user", "content": "讲解—下温故而知新"})
    response = model.chat(tokenizer, messages)
    print(response)
    

    主要修改如下:

    1. 导入了pipeline模块,用于处理模型的推理过程。
    2. AutoModelAutoTokenizer的导入语句修改为正确的格式。
    3. 删除了不必要的注释和变量定义。
    4. messages.append的语法修改为字典格式。
    5. model.chat的参数传递方式修改为正确的格式。
    2023-12-04 14:08:35
    赞同 展开评论 打赏
  • 预发布,没提供示例代码,您在模型卡片提交反馈留言咨询。。此回答整理自钉钉群:魔搭ModelScope开发者联盟群 ①

    2023-12-04 08:06:22
    赞同 展开评论 打赏

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

相关电子书

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