[feat] add more llm backbones

Eureca2017 · Apr 19, 2024 · e83ba6e · e83ba6e
1 parent 9408e86
commit e83ba6e
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 0 deletions.
diff --git a/...kernel/llm_config/llama2-13b-chat-hf.json → ...ernel/llm_config/Llama-2-13b-chat-hf.json b/...kernel/llm_config/llama2-13b-chat-hf.json → ...ernel/llm_config/Llama-2-13b-chat-hf.json
diff --git a/src/llm_kernel/llm_config/Meta-Llama-3-70B-Instruct.json b/src/llm_kernel/llm_config/Meta-Llama-3-70B-Instruct.json
@@ -0,0 +1,5 @@
+{
+    "model_type": "causal_lm",
+    "open_sourced": true,
+    "model_name": "meta-llama/Meta-Llama-3-70B-Instruct"
+}
diff --git a/src/llm_kernel/llm_config/Meta-Llama-3-8B-Instruct.json b/src/llm_kernel/llm_config/Meta-Llama-3-8B-Instruct.json
@@ -0,0 +1,5 @@
+{
+    "model_type": "causal_lm",
+    "open_sourced": true,
+    "model_name": "meta-llama/Meta-Llama-3-8B-Instruct"
+}
diff --git a/src/llm_kernel/llm_config/mistral-7b-it.json → .../llm_config/Mistral-7B-Instruct-v0.2.json b/src/llm_kernel/llm_config/mistral-7b-it.json → .../llm_config/Mistral-7B-Instruct-v0.2.json
diff --git a/src/llm_kernel/llm_config/Mixtral-8x22B-Instruct-v0.1.json b/src/llm_kernel/llm_config/Mixtral-8x22B-Instruct-v0.1.json
@@ -0,0 +1,5 @@
+{
+    "model_type": "causal_lm",
+    "open_sourced": true,
+    "model_name": "mistralai/Mixtral-8x22B-Instruct-v0.1"
+}
diff --git a/...lm_kernel/llm_config/mixtral-8x7b-it.json → ...lm_config/Mixtral-8x7B-Instruct-v0.1.json b/...lm_kernel/llm_config/mixtral-8x7b-it.json → ...lm_config/Mixtral-8x7B-Instruct-v0.1.json
diff --git a/src/llm_kernel/llm_config/gemma-7b-it.json → ...lm_kernel/llm_config/gemma-1.1-7b-it.json b/src/llm_kernel/llm_config/gemma-7b-it.json → ...lm_kernel/llm_config/gemma-1.1-7b-it.json