How to use from the
Use from the
PEFT library
from peft import PeftModel
from transformers import AutoModelForCausalLM

base_model = AutoModelForCausalLM.from_pretrained("/kaggle/working/models/base/Salesforce__codegen-350M-multi")
model = PeftModel.from_pretrained(base_model, "codegenstudio/codegen-350M-sql2nosql-lora")

codegenstudio/codegen-350M-sql2nosql-lora

LoRA adapter for sql2nosql on Salesforce/codegen-350M-multi.

  • Checkpoint version: v3
  • Task: sql2nosql

Usage

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

base = "Salesforce/codegen-350M-multi"
adapter = "codegenstudio/codegen-350M-sql2nosql-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.

Downloads last month
48
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for codegenstudio/codegen-350M-sql2nosql-lora

Adapter
(13)
this model