Modelpublic

care2achieve/codegen-350M-text2sql-lora

sourceHugging Faceupdated 2mo agoView on Hugging Face

0likes8downloads

Model Card

care2achieve/codegen-350M-text2sql-lora

LoRA adapter for text2sql on Salesforce/codegen-350M-multi.

—Checkpoint version: v4
—Task: text2sql

Usage

python

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

base = "Salesforce/codegen-350M-multi"
adapter = "care2achieve/codegen-350M-text2sql-lora"

tokenizer = AutoTokenizer.from_pretrained(base)
model = AutoModelForCausalLM.from_pretrained(base)
model = PeftModel.from_pretrained(model, adapter)

Or use the multi-adapter API in this project's hf-deploy/ package.