b-mc2/sql-create-context
Viewer • Updated • 78.6k • 8.81k • 506
How to use AmbikaSoni/qwen2.5-coder-1.5b-sql-lora-v2c with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-1.5B-Instruct")
model = PeftModel.from_pretrained(base_model, "AmbikaSoni/qwen2.5-coder-1.5b-sql-lora-v2c")QLoRA adapter for Qwen/Qwen2.5-Coder-1.5B-Instruct fine-tuned on b-mc2/sql-create-context for text-to-SQL.
Qwen/Qwen2.5-Coder-1.5B-Instruct| Run | Baseline | Fine-tuned | Delta |
|---|---|---|---|
| V1 (Qwen2.5-1.5B-Instruct, r=16, α=32, 2500) | 42% | 61% | +19 |
| V2C (Qwen/Qwen2.5-Coder-1.5B-Instruct, r=16, α=32, 5000) | 35.0% | 70.0% | +35.0 |
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
import torch
bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_quant_type="nf4",
bnb_4bit_compute_dtype=torch.bfloat16,
bnb_4bit_use_double_quant=True)
base = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-1.5B-Instruct",
quantization_config=bnb, device_map="auto")
model = PeftModel.from_pretrained(base, "AmbikaSoni/qwen2.5-coder-1.5b-sql-lora-v2c")
tok = AutoTokenizer.from_pretrained("AmbikaSoni/qwen2.5-coder-1.5b-sql-lora-v2c")
Base model
Qwen/Qwen2.5-1.5B