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

请问下Modelscope /baichuan-7B/ 用cpu跑,是这样设置吗?

请问下Modelscope /baichuan-7B/ 用cpu跑,是这样设置吗?
text_generation_zh = pipeline(task=Tasks.text_generation, device='cpu', model='baichuan-inc/baichuan-7B',model_revision='v1.0.2')

展开
收起
青城山下庄文杰 2023-07-22 09:00:18 258 0
3 条回答
写回答
取消 提交回答
  • 北京阿里云ACE会长

    Modelscope /baichuan-7B/ 用CPU运行,设置应该是:

    python
    Copy
    from transformers import pipeline

    text_generation_zh = pipeline(
    task='text_generation',
    model='baichuan-inc/baichuan-7B',
    model_revision='v1.0.2',
    device=0 # 设置device为0表示使用CPU
    )
    也就是需要将device参数设置为0,表示使用CPU。

    你的代码:

    python
    Copy
    text_generation_zh = pipeline(task=Tasks.text_generation,
    device='cpu',
    model='baichuan-inc/baichuan-7B',
    model_revision='v1.0.2')
    有几点问题:

    Tasks.text_generation 应该是'text_generation'

    device参数应该设为整数0,表示CPU

    更正后的代码应该是:

    python
    Copy
    text_generation_zh = pipeline(
    task='text_generation',
    device=0,
    model='baichuan-inc/baichuan-7B',
    model_revision='v1.0.2'
    )
    希望以上解答可以帮助您正确设置使用CPU运行/baichuan-7B/ 这个模型。

    2023-07-27 18:45:29
    赞同 展开评论 打赏
  • 是的,您可以通过设置device='cpu'来在 ModelScope 中使用 CPU 进行推理。这将指示 ModelScope 在没有 GPU 支持的情况下使用 CPU 资源进行计算。

    对于使用 /baichuan-7B/ 模型进行文本生成任务(text generation)的示例代码:

    from modelscope import Tasks, pipeline
    
    text_generation_zh = pipeline(task=Tasks.text_generation, device='cpu', model='baichuan-inc/baichuan-7B', model_revision='v1.0.2')
    

    在上述示例中,device='cpu' 参数告诉 ModelScope 使用 CPU 进行推理。确保已安装正确的依赖项,并且系统具备足够的 CPU 计算资源来支持模型的运行。

    请注意,相对于使用 GPU 进行推理,使用 CPU 进行推理可能会导致性能较低和速度较慢,尤其是对于大型模型和复杂任务。但是,在没有 GPU 可用或需要限制资源使用的情况下,使用 CPU 仍然是一个可iable 的选择。

    2023-07-24 13:21:04
    赞同 展开评论 打赏
  • 是的,要在Modelscope上使用/baichuan-7B/模型并使用CPU进行推理,需要进行以下设置:

    1. 在Modelscope上创建一个新项目或选择一个现有项目。
    2. 在项目页面的左侧导航栏中,选择"模型"选项卡。
    3. 在模型页面上,点击"添加模型"按钮。
    4. 在模型配置页面上,选择模型源为"Hugging Face"。
    5. 在模型名称字段中输入"baichuan-7B"。
    6. 在模型代码库字段中输入"Hugging Face"。
    7. 在模型代码字段中输入"baichuan-7B"。
    8. 在模型参数字段中输入"cpu",以指定使用CPU进行推理。
    9. 点击"保存"按钮以保存模型配置。
    10. 现在,您可以使用Modelscope上的/baichuan-7B/模型,并使用CPU进行推理了。
    2023-07-22 19:44:04
    赞同 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
RDS SQL Server CPU高使用率性能优化 立即下载