---
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.