care2achieve's picture
Upload folder using huggingface_hub
902472f verified
|
Raw
History Blame Contribute Delete
783 Bytes
---
library_name: peft
base_model: Salesforce/codegen-350M-multi
tags:
- lora
- peft
- code
- text2sql
- codegen
pipeline_tag: text-generation
---
# care2achieve/codegen-350M-text2sql-lora
LoRA adapter for **text2sql** on [Salesforce/codegen-350M-multi](https://huggingface.co/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.