模型加载常见问题

safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

问题代码:

model = AutoModelForVision2Seq.from_pretrained(
    "/data-nvme/yang/Qwen2.5-VL-32B-Instruct",
    trust_remote_code=True,
    torch_dtype=torch.bfloat16,
    device_map="auto",
    max_memory={i: '24GiB' for i in range(2)}
)

解决办法:

使用Git LFS完整克隆

cd /data-nvme/yang/
git lfs install
git clone https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct 
cd /data-nvme/yang/Qwen2.5-VL-7B-Instruct
git lfs pull --include="*.safetensors,*.json,*.py"

使用huggingface模型对应的调试代码,调试通过;

样例图片地址:

https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg

调色代码输出:

['这张图片展示了一位女士和一只狗在海滩上互动的场景。女士坐在沙滩上,穿着格子衬衫和牛仔裤,面带微笑,与狗进行握手。狗戴着彩色的项圈,看起来非常友好。背景是平静的海浪和渐变的天空,阳光洒在沙滩上,营造出一种温馨和谐的氛围。']

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值