专门给fastvllm的模型,转换好的模型。chatglm3-6b-fp16的flm文件,名字命名错

我要开发同款
匿名用户2024年07月31日
33阅读
所属分类aiPytorch
开源地址https://modelscope.cn/models/flyiot/fastllm-chatglm2-6b-fp16-flm
授权协议Apache License 2.0

作品详情

Clone with HTTP

快速下载方法:

 git clone https://www.modelscope.cn/flyiot/fastllm-chatglm2-6b-fp16-flm.git

模型是根据fastllm 的工具生成的。

修改 /fastllm/build/tools/chatglm_export.py 修改模型路径:

import sys
from transformers import AutoTokenizer, AutoModel
from fastllm_pytools import torch2flm

if __name__ == "__main__":
    tokenizer = AutoTokenizer.from_pretrained("/root/autodl-tmp/chatglm3-6b", trust_remote_code=True)
    model = AutoModel.from_pretrained("/root/autodl-tmp/chatglm3-6b", trust_remote_code=True)
    model = model.eval()

...
cd build
python3 tools/chatglm_export.py chatglm2-6b-fp16.flm float16 #导出float16模型

即可生成模型。

具体方法: https://blog.csdn.net/freewebsys/article/details/134816095

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论