add yi 6B

app.py

@@ -97,6 +97,8 @@ def respond(
         _model_name = "Qwen/Qwen1.5-72B"
     elif model_name == "Yi-34B":
         _model_name = "zero-one-ai/Yi-34B"
+    elif model_name == "Yi-6B":
+        _model_name = "zero-one-ai/Yi-6B"
     elif model_name == "OLMO":
         _model_name = "allenai/OLMo-7B"
     else:
@@ -170,7 +172,9 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label) as demo:
 
                         **Talk with __BASE__ LLMs which are not fine-tuned at all.**
                         """)
-            model_name = gr.Radio(["Llama-3-8B", "Llama-3-70B", "Mistral-7B-v0.1", "Mixtral-8x22B", "Yi-34B", "Llama-2-7B", "Llama-2-70B", "OLMO"], value="Llama-3-8B", label="Base LLM name")
+            model_name = gr.Radio(["Llama-3-8B", "Llama-3-70B", "Mistral-7B-v0.1", 
+                                   "Mixtral-8x22B", "Yi-6B", "Yi-34B", "Llama-2-7B", "Llama-2-70B", "OLMO"]
+                                  , value="Llama-3-8B", label="Base LLM name")
         with gr.Column():
             together_api_key = gr.Textbox(label="🔑 Together APIKey", placeholder="Enter your Together API Key. Leave it blank if you want to use the default API key.", type="password", elem_id="api_key")
             with gr.Column():