利用Intel transformers拓展包加速CPU大模型推理

  1. 技术介绍

    Intel Extension for Transformers 使用了多种优化技术来提升深度学习模型的性能。主要包括:
    1. 量化(Quantization):通过减少模型参数的位数来减小模型大小并加速推理过程。提供了包括AWQ,SmoothQuant在内的先进量化技术;
    2. 混合精度(Mixed Precision):使用不同的数据类型(如bfloat16、int8)来平衡计算速度和精度;
    3. 与Intel CPU的SIMD指令结合加速推理;
    4. FP8数据类型:使用8位浮点数来进一步压缩模型;
    5. 自动量化(AutoQuant):自动化量化过程,以找到最佳的量化策略;
    6. FlashAttention:一种优化注意力机制的方法,用于加速长序列的处理;
  2. 实验过程

    利用ModelScope提供的免费CPU算力尝试ITREX的性能。步骤如下:
    1. 在ModelScope上注册账号并绑定阿里云账号后,点击左侧创建Notebook,可以选择我们的计算环境;
    2. 选择CPU环境,启动Notebook后点击查看Notebook,就可以进入工作空间。
    3. 启动一个终端,在终端中输入下面的命令来下载ITREX所需的conda环境
    4. 再输入下面的命令安装环境和对应的ipykernel。
    5. 环境准备完毕,下面开始下载模型,我们选择的是zhipuAI的ChatGLM3模型和一个embedding模型
    6. 这次试验我们计划试一下RAG插件。首先准备一条简短的知识,格式为json,保存为文件sample.json
    7. 执行下面的代码加载模型和插件
  3. 推理结果        ​​​   

    可以看到,关闭retrieval插件时模型的回答是错误的,打开后模型输出了正确答案。
  4. 总结

    ITREX提供了一个在Intel CPU上进行高效推理的方案,不仅支持多种前沿的Efficient inference技术,并且与Intel硬件结合地很好。
  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值