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

ModelScope这个如何指定多卡加载,如指定4、5两张卡加载?

ModelScope这个如何指定多卡加载,如指定4、5两张卡加载?
from modelscope import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(model_path, device_map="auto")用modelscope框架加载大模型,不应该是同样的方式么,还区分模型?我加载的是Qwen1.5-72B-Chat-GPTQ-Int4

展开
收起
夹心789 2024-06-09 08:25:17 17 0
1 条回答
写回答
取消 提交回答
  • 您加个环境变量,设置4,5两张卡 ,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-06-11 09:36:19
    赞同 展开评论 打赏

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

相关电子书

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