databricks/databricks-dolly-15k
Viewer • Updated • 15k • 35.1k • 972
How to use ArgusForge/qwen2.5-14b-dolly-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("/workspace/qwen2.5-14b")
model = PeftModel.from_pretrained(base_model, "ArgusForge/qwen2.5-14b-dolly-lora")A small LoRA adapter for Qwen2.5-14B-Instruct, trained on databricks-dolly-15k. This is a demonstration artifact. It exists to show a full open-model loop end to end (fine-tune, merge, quantize, serve), not to be a state-of-the-art model.
Trained on dolly's concise, human-written responses, the adapter shifts the base model away from its long multi-section default toward shorter, more direct answers.
Prompt: Explain why the sky is blue to a ten year old.
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
import torch
base = "Qwen/Qwen2.5-14B-Instruct"
tok = AutoTokenizer.from_pretrained(base)
model = AutoModelForCausalLM.from_pretrained(base, dtype=torch.bfloat16, device_map="auto")
model = PeftModel.from_pretrained(model, "ArgusForge/qwen2.5-14b-dolly-lora")