Distilling LLM Agent into Small Models with Retrieval and Code Tools
Paper • 2505.17612 • Published • 81
How to use agent-distillation/agent_distilled_Qwen2.5-3B-Instruct with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-3B-Instruct")
model = PeftModel.from_pretrained(base_model, "agent-distillation/agent_distilled_Qwen2.5-3B-Instruct")This model is distilled Qwen2.5-3B-Instruct from agent trajectories in this dataset.