Add support for sqlcoder-34b-alpha #4

Author: pramitchoudhary

sidekick/utils.py

@@ -23,6 +23,7 @@
 
 MODEL_CHOICE_MAP_EVAL_MODE = {
     "h2ogpt-sql-sqlcoder2": "defog/sqlcoder2",
+    "h2ogpt-sql-sqlcoder-34b-alpha": "defog/sqlcoder-34b-alpha",
     "h2ogpt-sql-nsql-llama-2-7B": "NumbersStation/nsql-llama-2-7B",
     "gpt-3.5-turbo": "gpt-3.5-turbo-1106",
     "gpt-4-8k": "gpt-4",
@@ -32,6 +33,7 @@
 
 MODEL_CHOICE_MAP_DEFAULT = {
     "h2ogpt-sql-sqlcoder2": "defog/sqlcoder2",
+    "h2ogpt-sql-sqlcoder-34b-alpha": "defog/sqlcoder-34b-alpha",
     "h2ogpt-sql-nsql-llama-2-7B": "NumbersStation/nsql-llama-2-7B"
 
 }

start.py

@@ -26,6 +26,9 @@ def setup_dir(base_path: str):
 print(f"Download model 2...")
 snapshot_download(repo_id="defog/sqlcoder2", cache_dir=f"{base_path}/models/")
 
+print(f"Download model 3...")
+snapshot_download(repo_id="defog/sqlcoder-34b-alpha", cache_dir=f"{base_path}/models/")
+
 print(f"Download embedding model...")
 snapshot_download(repo_id="BAAI/bge-base-en", cache_dir=f"{base_path}/models/sentence_transformers/")