{ "_name_or_path": "llmware/slim-sql-qwen-base", "aib_version": "model_archive_083024-qwen2-15-base-sql-25k-eot-train-sql2-1", "architectures": [ "Qwen2ForCausalLM" ], "attention_dropout": 0.0, "bos_token_id": 151643, "eos_token_id": 151643, "hidden_act": "silu", "hidden_size": 1536, "initializer_range": 0.02, "intermediate_size": 8960, "max_position_embeddings": 131072, "max_window_layers": 28, "model_type": "qwen2", "num_attention_heads": 12, "num_hidden_layers": 28, "num_key_value_heads": 2, "rms_norm_eps": 1e-06, "rope_theta": 1000000.0, "sliding_window": null, "tie_word_embeddings": true, "trained": "custom training", "training_comments": "qwen-base-15-sql-25k-eot-1", "training_dataset": [ "train_sql_big2.jsonl" ], "training_timestamp": "Fri Aug 30 19:21:20 2024", "transformers_version": "4.43.4", "use_cache": true, "use_sliding_window": false, "vocab_size": 151936, "prompt_wrapper": "human_bot", "description": "slim-sql is a text-to-sql model.", "prompt_format": " {table_schema} \n {question} \n:", "output_format": "{sql}", "tokenizer_local": "tokenizer_tl.json", "tokenizer_config": {"bos_id":[1], "bos_token":[""], "eos_id":[2],"eos_token":[""]}, "model_parent": "llmware/slim-sql-1b-v0", "description": "Text-to-SQL model from llmware - finetuned on qwen - 3 parameter base", "quantization": "int4", "model_family": "OVGenerativeModel", "parameters": 3.0, "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" } ] }