jonathanjordan21 commited on
Commit
4196be5
1 Parent(s): 3ad64ff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
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 MySQL table schema.
29
- You can modify the table schema to match MySQL table schema if you are using different type of SQL database (e.g. PostgreSQL, Oracle, etc).
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 table_1341598_10 WHERE district = "Florida 18""""
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