b-mc2/sql-create-context
Viewer โข Updated โข 78.6k โข 5.79k โข 518
How to use krisrajz/text2sql-qwen25-1p5b-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct")
model = PeftModel.from_pretrained(base_model, "krisrajz/text2sql-qwen25-1p5b-lora")LoRA adapter for Qwen/Qwen2.5-1.5B-Instruct fine-tuned for text-to-SQL on b-mc2/sql-create-context.
Exact match on 200 held-out examples: base 0.015, fine-tuned 0.710.
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct")
model = PeftModel.from_pretrained(AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct"), "krisrajz/text2sql-qwen25-1p5b-lora")
The prompt template is defined in the training repository (src/prompt.py).