jonathanjordan21
commited on
Commit
•
4196be5
1
Parent(s):
3ad64ff
Update README.md
Browse files
README.md
CHANGED
@@ -25,8 +25,8 @@ This model is based on the declare-lab/flan-alpaca-base model finetuned with wik
|
|
25 |
|
26 |
## Uses
|
27 |
|
28 |
-
The model generates a string of SQL query based on a question and
|
29 |
-
|
30 |
The generated SQL query can be run perfectly on the python SQL connection (e.g. psycopg2, mysql_connector, etc).
|
31 |
|
32 |
#### Limitations
|
@@ -42,7 +42,7 @@ The generated SQL query can be run perfectly on the python SQL connection (e.g.
|
|
42 |
```
|
43 |
### Output Example
|
44 |
```python
|
45 |
-
"""SELECT * FROM
|
46 |
```
|
47 |
|
48 |
### How to use
|
@@ -59,7 +59,6 @@ model_ = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, r
|
|
59 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
60 |
|
61 |
model = PeftModel.from_pretrained(model_, model_id)
|
62 |
-
model = get_peft_model(model,config)
|
63 |
```
|
64 |
|
65 |
Model inference
|
|
|
25 |
|
26 |
## Uses
|
27 |
|
28 |
+
The model generates a string of SQL query based on a question and table columns. **The generated query always uses "table" as the table name**.
|
29 |
+
Feel free to change the table name in the generated query to match your actual SQL table.
|
30 |
The generated SQL query can be run perfectly on the python SQL connection (e.g. psycopg2, mysql_connector, etc).
|
31 |
|
32 |
#### Limitations
|
|
|
42 |
```
|
43 |
### Output Example
|
44 |
```python
|
45 |
+
"""SELECT * FROM table WHERE district = "Florida 18""""
|
46 |
```
|
47 |
|
48 |
### How to use
|
|
|
59 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
60 |
|
61 |
model = PeftModel.from_pretrained(model_, model_id)
|
|
|
62 |
```
|
63 |
|
64 |
Model inference
|