search

SAI990323 / BIGRec

50 stars 5 forks source link

safetensors_rust.SafetensorError: Error while deserializing header: InvalidHeaderDeserialization #8

Open lingaoqiming123 opened 6 months ago

lingaoqiming123 commented 6 months ago

During inference, an error occurs when merging the base_model and lora_weights. safetensors_rust.SafetensorError: Error while deserializing header: InvalidHeaderDeserialization image

lingaoqiming123 commented 6 months ago

已解决。在4.39.3的transfomers中save_safetensors默认值为True。只需在train.py中args=transformers.TrainingArguments(）处中添加一行save_safetensors=False。

image