【踩坑日记15】safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

一只_程序媛

已于 2024-04-15 19:21:39 修改

阅读量2.1k

点赞数 4

文章标签：开发语言后端

于 2024-04-14 14:32:07 首次发布

本文链接：https://blog.csdn.net/weixin_47894469/article/details/137744216

版权

问题描述

加载stabilityai/stable-diffusion-xl-base-1.0模型时，出现问题。

Error while deserializing header: HeaderTooLarge
  File "/home/XXX/code/dreambooth_lora/train_dreambooth_lora_sdxl_advanced.py", line 1278, in main
    text_encoder_two = text_encoder_cls_two.from_pretrained(
  File "/home/XXX/code/dreambooth_lora/train_dreambooth_lora_sdxl_advanced.py", line 2366, in <module>
    main(args)
safetensors_rust.SafetensorError: Error while deserializing header: HeaderTooLarge

问题分析

下载的权重中textencoder的.safetensor文件可能出问题了。

问题解决

进入stabilityai/stable-diffusion-xl-base-1.0主页，重新下载textencode.safetensor文件，即下载文件text_encoder_2/model.fp16.safetensors和文件text_encoder_2/model.safetensors。