File size: 751 Bytes
29afefb 21394e2 29afefb 21394e2 29afefb 21394e2 29afefb 21394e2 29afefb 21394e2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | ---
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.
|