sft-kl-lora-trainer. Custom trl.SFTTrainer that adds a KL divergence loss between a LoRA-adapted model and its base model.

github.com/BY571/sft-kl-lora-trainer

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.