FastChat icon indicating copy to clipboard operation
FastChat copied to clipboard

How to save lora weights only???

Open lucasjinreal opened this issue 2 years ago • 1 comments

Currently the weights saved are all models weights include optimzier, is HUGE,

how to set save lora weights only?

image

lucasjinreal avatar May 31 '23 12:05 lucasjinreal

Current checkpoint from huggingface/deepspeed does not support this functionality. To store only the LoRA weight after the whole training progress, see here. Maybe one can monkey patch the hf trainer/deepspeed's checkpoint function to do the same thing

ZYHowell avatar May 31 '23 18:05 ZYHowell