Add code llama info (lm-sys#2331)

wangzhen263 · Aug 28, 2023 · ce3311d · ce3311d
1 parent 106670d
commit ce3311d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py
@@ -61,6 +61,12 @@ def get_model_info(name: str) -> ModelInfo:
     "https://ai.meta.com/llama/",
     "open foundation and fine-tuned chat models by Meta",
 )
+register_model_info(
+    ["codellama-34b-instruct", "codellama-13b-instruct", "codellama-7b-instruct"],
+    "Code Llama",
+    "https://ai.meta.com/blog/code-llama-large-language-model-coding/",
+    "open foundation models for code by Meta",
+)
 register_model_info(
     [
         "vicuna-33b",

diff --git a/fastchat/serve/gradio_block_arena_anony.py b/fastchat/serve/gradio_block_arena_anony.py
@@ -170,11 +170,13 @@ def share_click(state0, state1, model_selector0, model_selector1, request: gr.Re
     "palm-2": 1.5,
     "llama-2-70b-chat": 1.5,
     "llama-2-13b-chat": 1.5,
+    "codellama-34b-instruct": 1.5,
     "vicuna-33b": 1.5,
     "vicuna-13b": 1.5,
     "mpt-30b-chat": 1.5,
     "wizardlm-13b": 1.5,
     # tier 2
+    "codellama-13b-instruct": 1.0,
     "guanaco-33b": 1.0,
     "vicuna-7b": 1.0,
     "llama-2-7b-chat": 1.0,