Update config.json

config.json

@@ -1,35 +1,109 @@
-{
-  "_name_or_path": "llmware/slim-sql-phi-3",
-  "architectures": [
-    "Phi3ForCausalLM"
-  ],
-  "attention_bias": false,
-  "attention_dropout": 0.0,
-  "auto_map": {
-    "AutoConfig": "llmware/slim-sql-phi-3--configuration_phi3.Phi3Config",
-    "AutoModelForCausalLM": "llmware/slim-sql-phi-3--modeling_phi3.Phi3ForCausalLM"
-  },
-  "bos_token_id": 1,
-  "embd_pdrop": 0.0,
-  "eos_token_id": 32000,
-  "hidden_act": "silu",
-  "hidden_size": 3072,
-  "initializer_range": 0.02,
-  "intermediate_size": 8192,
-  "max_position_embeddings": 4096,
-  "model_type": "phi3",
-  "num_attention_heads": 32,
-  "num_hidden_layers": 32,
-  "num_key_value_heads": 32,
-  "original_max_position_embeddings": 4096,
-  "pad_token_id": 32000,
-  "resid_pdrop": 0.0,
-  "rms_norm_eps": 1e-05,
-  "rope_scaling": null,
-  "rope_theta": 10000.0,
-  "sliding_window": 2047,
-  "tie_word_embeddings": false,
-  "transformers_version": "4.43.4",
-  "use_cache": true,
-  "vocab_size": 32064
-}
+{
+  "_name_or_path": "llmware/slim-sql-phi-3",
+  "architectures": [
+    "Phi3ForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "llmware/slim-sql-phi-3--configuration_phi3.Phi3Config",
+    "AutoModelForCausalLM": "llmware/slim-sql-phi-3--modeling_phi3.Phi3ForCausalLM"
+  },
+  "bos_token_id": 1,
+  "embd_pdrop": 0.0,
+  "eos_token_id": 32000,
+  "hidden_act": "silu",
+  "hidden_size": 3072,
+  "initializer_range": 0.02,
+  "intermediate_size": 8192,
+  "max_position_embeddings": 4096,
+  "model_type": "phi3",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "num_key_value_heads": 32,
+  "original_max_position_embeddings": 4096,
+  "pad_token_id": 32000,
+  "resid_pdrop": 0.0,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "sliding_window": 2047,
+  "tie_word_embeddings": false,
+  "transformers_version": "4.43.4",
+  "use_cache": true,
+  "vocab_size": 32064,
+  "prompt_wrapper": "human_bot",
+  "description": "slim-sql is a text-to-sql model.",
+  "prompt_format": "<human> {table_schema} \n {question} \n<bot>:",
+  "output_format": "{sql}",
+  "tokenizer_local": "tokenizer_phi3.json",
+  "tokenizer_config": {"bos_id":[1], "bos_token":["<s>"], "eos_id":[2],"eos_token":["</s>"]},
+  "model_parent": "llmware/slim-sql-phi-3",
+  "description": "Text-to-SQL model from llmware - finetuned on phi-3 - 3.8 parameter base",
+  "quantization": "int4",
+  "model_family": "OVGenerativeModel",
+  "parameters": 3.8,
+  "primary_keys": ["sql"],
+  "output_values": ["{{sql statement}}"],
+  "publisher": "llmware",
+  "release_date": "september 2024",
+  "function_call": "sql",
+  "test_params": ["sql"],
+   "test_set": [
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "Which customers are VIP customers?",
+   "answer": "SELECT * FROM customers WHERE vip_customer='yes'"
+  },
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "What is the annual spend for customer Rachel Michaels?",
+   "answer": "SELECT annual_spend FROM customers WHERE customer_name='Rachel Michaels'"
+  },
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "How many customers spend more than $1000 per year?",
+   "answer": "SELECT COUNT(*) FROM customers WHERE annual_spend > $1000"
+  },
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "Who are the customers with gold customer level?",
+   "answer": "SELECT customer_name FROM customers WHERE customer_level = 'gold'"
+  },
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "Which customer has account number 9382035?",
+   "answer": "SELECT * FROM customers WHERE account_number = 9382035"
+  },
+  {
+   "context": "CREATE TABLE customers (customer_name text, account_number integer,customer_level text, vip_customer text, annual_spend text, user_name text)",
+   "query": "What is the account number of customer Susanna Jones?",
+   "answer": "SELECT account_number FROM customers WHERE customer_name='Susanna Jones'"
+  },
+  {
+   "context": "CREATE TABLE library (library_name text, unique_doc_id integer, documents integer, blocks integer, images integer, pages integer, tables integer, account_name text)",
+   "query": "How many pages are in the human resources library?",
+   "answer": "SELECT pages FROM library WHERE library_name = 'human resources'"
+  },
+  {
+   "context": "CREATE TABLE library (library_name text, unique_doc_id integer, documents integer, blocks integer, images integer, pages integer, tables integer, account_name text)",
+   "query": "Which libraries have more than 1000 images?",
+   "answer": "SELECT * FROM library WHERE images > 1000"
+  },
+  {
+   "context": "CREATE TABLE library (library_name text, unique_doc_id integer, documents integer, blocks integer, images integer, pages integer, tables integer, account_name text)",
+   "query": "How many blocks are in the finance library?",
+   "answer": "SELECT blocks FROM library WHERE library_name = 'finance library'"
+  },
+  {
+   "context": "CREATE TABLE library (library_name text, unique_doc_id integer, documents integer, blocks integer, images integer, pages integer, tables integer, account_name text)",
+   "query": "What is a list of all of the libraries?",
+   "answer": "SELECT * FROM library"
+  },
+  {
+   "context": "CREATE TABLE library (library_name text, unique_doc_id integer, documents integer, blocks integer, images integer, pages integer, tables integer, account_name text)",
+   "query": "Which library has unique_doc_id of 8329?",
+   "answer": "SELECT * FROM library WHERE unique_doc_id = 8329"
+  }
+ ]
+}