From 72414b951cc4f81c5e737810e16957d8975da1c0 Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Mon, 11 Dec 2023 21:38:45 -0500
Subject: [PATCH] x

---
 langchain_benchmarks/model_registration.py | 107 +++++++++++++++++++++
 1 file changed, 107 insertions(+)

diff --git a/langchain_benchmarks/model_registration.py b/langchain_benchmarks/model_registration.py
index 6df93e7d..2a9c4606 100644
--- a/langchain_benchmarks/model_registration.py
+++ b/langchain_benchmarks/model_registration.py
@@ -147,6 +147,113 @@
             "model": "accounts/fireworks/models/llama-v2-70b-chat",
         },
     ),
+    RegisteredModel(
+        provider="fireworks",
+        name='mixtral-moe-8x7b-chat-fw',
+        type="chat",
+        description="Mistral MoE model, unofficial implementation. Further fine-tuned for chat by Fireworks.",
+        params={
+            "model": "accounts/fireworks/models/mixtral-moe-8x7b-chat",
+        },
+    ),
+    RegisteredModel(
+        provider="fireworks",
+        name="llama-v2-7b-llm-fw",
+        type="llm",
+    )
+
+    {
+        "name": "Mixtral MoE 8x7B",
+        "description": "Mistral MoE model, unofficial implementation.",
+        "type": "llm",
+    },
+    {
+        "name": "Capybara 34B",
+        "description": "34B chat model from NousResearch, based on Yi-34B-200k.",
+        "type": "chat",
+    },
+    {
+        "name": "Yi 34B 200k context window",
+        "description": "34B LLM model from 01.ai, with context window 200k.",
+        "params": {"model": "accounts/fireworks/models/yi-34b-200k-capybara"},
+        "type": "llm",
+    },
+    {
+        "name": "Yi 6B",
+        "description": "6B LLM model from 01.ai.",
+        "params": {"model": "accounts/fireworks/models/yi-6b"},
+        "type": "llm",
+    },
+]
+
+
+[
+    {
+        "name": "Mistral 7B Instruct",
+        "description": "Mistral-7B model fine-tuned for conversations.",
+        "params": {"model": "accounts/fireworks/models/mistral-7b-instruct"},
+        "type": "llm",
+    },
+    {
+        "name": "Llama 2 13B code instruct",
+        "description": "Instruction-tuned version of Llama 2 13B, optimized for code generation.",
+        "params": {"model": "accounts/fireworks/models/llama-2-13b-code-instruct"},
+        "type": "llm",
+    },
+    {
+        "name": "Llama 2 34B Code Llama instruct",
+        "description": "Code Llama 34B, optimized for code generation.",
+        "params": {"model": "accounts/fireworks/models/llama-2-34b-code-instruct"},
+        "type": "llm",
+    },
+    {
+        "name": "Llama 2 7B Chat",
+        "description": "Fine-tuned version of Llama 2 7B, optimized for dialogue applications using RLHF, comparable to ChatGPT.",
+        "params": {"model": "accounts/fireworks/models/llama-2-7b-chat"},
+        "type": "chat",
+    },
+    {
+        "name": "Llama 2 13B Chat",
+        "description": "Fine-tuned version of Llama 2 13B, optimized for dialogue applications using RLHF, comparable to ChatGPT.",
+        "params": {"model": "accounts/fireworks/models/llama-2-13b-chat"},
+        "type": "chat",
+    },
+    {
+        "name": "Llama 2 70B Chat",
+        "description": "Fine-tuned version of Llama 2 70B, optimized for dialogue applications using RLHF, comparable to ChatGPT.",
+    },
+    {
+        "name": "StarCoder 7B",
+        "description": "7B parameter model trained on 80+ programming languages from The Stack (v1.2), using Multi Query Attention and Fill-in-the-Middle objective.",
+    },
+    {
+        "name": "StarCoder 15.5B",
+        "description": "15.5B parameter model trained on 80+ programming languages from The Stack (v1.2), using Multi Query Attention and Fill-in-the-Middle objective.",
+    },
+    {
+        "name": "Traditional Chinese Llama2 QLoRa",
+        "description": "Fine-tuned Llama 2 model on traditional Chinese Alpaca dataset.",
+    },
+    {
+        "name": "Llama 2 13B French",
+        "description": "Fine-tuned meta-llama/Llama-2-13b-chat-hf to answer French questions in French.",
+    },
+    {
+        "name": "Chinese Llama 2 LoRA 7B",
+        "description": "The LoRA version of Chinese-Llama-2 based on Llama-2-7b-hf.",
+    },
+    {
+        "name": "Bleat",
+        "description": "Enables function calling in LLaMA 2, similar to OpenAI's implementation for ChatGPT.",
+    },
+    {
+        "name": "Llama2 13B Guanaco QLoRA GGML",
+        "description": "Fine-tuned Llama 2 13B model using the Open Assist dataset.",
+    },
+    {
+        "name": "Llama 7B Summarize",
+        "description": "Summarizes articles and conversations.",
+    },
 ]
 
 _ANTHROPIC_MODELS = [