请问ModelScope千问72B两张80G的卡,加载不成功为啥呢?

"请问ModelScope千问72B两张80G的卡,加载不成功为啥呢? oom model = AutoModelForCausalLM.from_pretrained(""/data1/plm/Qwen-72B"", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map=""auto"").eval().to(""cuda"")

"

展开
收起
小小爱吃香菜 2024-04-02 21:03:09 113 分享 版权
1 条回答
写回答
取消 提交回答

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

热门讨论

热门文章

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