初始化项目,由ModelHub XC社区提供模型
Model: OneGate/OG-SQL-7B Source: Original Platform
This commit is contained in:
77
README.md
Normal file
77
README.md
Normal file
@@ -0,0 +1,77 @@
|
||||
---
|
||||
license: cc-by-4.0
|
||||
language:
|
||||
- en
|
||||
tags:
|
||||
- Text-to-sql
|
||||
---
|
||||
# OGSQL-7B
|
||||
|
||||

|
||||
|
||||
### Model Description
|
||||
OGSQL-7B was fine-tuned for the task of converting natural language text into SQL queries.
|
||||
|
||||
- **Model type**: Transformer
|
||||
- **Language(s) (NLP)**: SQL (target language for generation)
|
||||
- **Finetuned from model**: codellama 7b
|
||||
|
||||
## Use Case
|
||||
OGSQL-7B is designed to facilitate the conversion of natural language queries into structured SQL commands, aiding in database querying without the need for manual SQL knowledge.
|
||||
|
||||
## How to Get Started with the Model
|
||||
```python
|
||||
# Example code to load and use the model
|
||||
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
||||
|
||||
model_name = "OGSQL-7B"
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
||||
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
||||
|
||||
def generate_sql(query):
|
||||
inputs = tokenizer.encode(query, return_tensors="pt")
|
||||
outputs = model.generate(inputs)
|
||||
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
||||
|
||||
# Example use
|
||||
query = """
|
||||
using this context:
|
||||
-- Create Customers Table
|
||||
CREATE TABLE Customers (
|
||||
customer_id INTEGER PRIMARY KEY,
|
||||
name TEXT NOT NULL,
|
||||
email TEXT,
|
||||
join_date DATE
|
||||
);
|
||||
|
||||
-- Create Products Table
|
||||
CREATE TABLE Products (
|
||||
product_id INTEGER PRIMARY KEY,
|
||||
name TEXT NOT NULL,
|
||||
price DECIMAL(10, 2)
|
||||
);
|
||||
|
||||
-- Create Orders Table
|
||||
CREATE TABLE Orders (
|
||||
order_id INTEGER PRIMARY KEY,
|
||||
customer_id INTEGER,
|
||||
product_id INTEGER,
|
||||
order_date DATE,
|
||||
quantity INTEGER,
|
||||
total_price DECIMAL(10, 2),
|
||||
FOREIGN KEY (customer_id) REFERENCES Customers(customer_id),
|
||||
FOREIGN KEY (product_id) REFERENCES Products(product_id)
|
||||
);
|
||||
|
||||
show me all the orders from last month , sort by date
|
||||
|
||||
|
||||
"""
|
||||
print(generate_sql(query))
|
||||
|
||||
```
|
||||
|
||||
|
||||
## alternatively you can use this notebook:
|
||||
[](https://colab.research.google.com/drive/1zfuzV3R1GQflHV_va03WArb8vhwPh_2T)
|
||||
|
||||
Reference in New Issue
Block a user