Team Ai
Modelpublic

care2achieve/codegen-350M-text2sql-lora

sourceHugging Faceupdated 3mo agoView on Hugging Face
0likes17downloads
README.md35 linesDownload Raw Back to root
1---2library_name: peft3base_model: Salesforce/codegen-350M-multi4tags:5- lora6- peft7- code8- text2sql9- codegen10pipeline_tag: text-generation11---12 13# care2achieve/codegen-350M-text2sql-lora14 15LoRA adapter for **text2sql** on [Salesforce/codegen-350M-multi](https://huggingface.co/Salesforce/codegen-350M-multi).16 17- Checkpoint version: `v4`18- Task: `text2sql`19 20## Usage21 22```python23from peft import PeftModel24from transformers import AutoModelForCausalLM, AutoTokenizer25 26base = "Salesforce/codegen-350M-multi"27adapter = "care2achieve/codegen-350M-text2sql-lora"28 29tokenizer = AutoTokenizer.from_pretrained(base)30model = AutoModelForCausalLM.from_pretrained(base)31model = PeftModel.from_pretrained(model, adapter)32```33 34Or use the multi-adapter API in this project's `hf-deploy/` package.35