llama-cookbook
llama-cookbook copied to clipboard
Inference with "FULL_STATE_DICT" checkpoint from FSDP fine tuning
🚀 The feature, motivation and pitch
I can get a single checkpoint after using FSDP fine tune the model.
My question is how can I use this single checkpoint for inference?
Thanks
Alternatives
No response
Additional context
No response