add olmo

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 == "OLMO":
+        _model_name = "allenai/OLMo-7B"
     else:
         raise ValueError("Invalid model name")
     # _model_name = "meta-llama/Llama-3-8b-hf"
@@ -168,12 +170,12 @@ 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"], 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-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():
                 with gr.Row():
-                    max_tokens = gr.Textbox(value=1024, label="Max tokens")
+                    max_tokens = gr.Textbox(value=256, label="Max tokens")
                     temperature = gr.Textbox(value=0.5, label="Temperature")
             # with gr.Column():
             #     with gr.Row():