会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
Loading
bringlu
新随笔
联系
订阅
管理
2023年9月8日
利用 device_map、torch.dtype、bitsandbytes 压缩模型参数控制使用设备
摘要: 为了更好的阅读体验,请点击这里 device_map 以下内容参考 Huggingface Accelerate文档:超大模型推理方法 在 HuggingFace 中有个重要的关键字是 device_map,它可以简单控制模型层部署在哪些硬件上。 设置参数 device_map="auto",Acc
阅读全文
posted @ 2023-09-08 09:46 bringlu
阅读(15089)
评论(0)
推荐(2)
公告