Lightning-AI / litgpt

Pretrain, finetune, deploy 20+ LLMs on your own data. Uses state-of-the-art techniques: flash attention, FSDP, 4-bit, LoRA, and more.
https://lightning.ai
Apache License 2.0
6.85k stars 726 forks source link

Qwen1.5 Family Support #1340

Open junzhang-zj opened 3 weeks ago

junzhang-zj commented 3 weeks ago

I want to use litgpt to fine-tune the Qwen1.5 family model. Is it supported? Do I need to pass convert_hf_checkpoint.py first and then load it? But I reported an error in the first step, 'Qwen1.5-4B' is not a supported config name.