| base_model: Salesforce/codegen-350M-multi | |
| library_name: peft | |
| pipeline_tag: text-generation | |
| tags: | |
| - lora | |
| - peft | |
| - text2sql | |
| # codegenstudio/codegen-350M-text2sql-lora | |
| LoRA adapter for text2sql on Salesforce/codegen-350M-multi. | |
| - Checkpoint version: `v3` | |
| - Task: `text2sql` | |
| ## Usage | |
| ```python | |
| from peft import PeftModel | |
| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| base = "Salesforce/codegen-350M-multi" | |
| adapter = "codegenstudio/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 `fastapi-deploy/` package. | |