fix interface

app.py

@@ -197,49 +197,49 @@ def http_bot(state, temperature, top_p, max_new_tokens):
 
     if len(state.messages) == state.offset + 2:
         # First round of conversation
-
-        if "tinyllava" in model_name.lower():
-            if "3.1b" in model_name.lower() or "phi" in model_name.lower():
-                template_name = "phi"
-            elif "2.0b" in model_name.lower() or "stablelm" in model_name.lower():
-                template_name = "phi"
-            elif "qwen" in model_name.lower():
-                template_name = "qwen"
-            else:
-                template_name = "v1"
-
-        elif "llava" in model_name.lower():
-
-            if "llama-2" in model_name.lower():
-                template_name = "llava_llama_2"
-            elif "v1" in model_name.lower():
-                if "mmtag" in model_name.lower():
-                    template_name = "v1_mmtag"
-                elif (
-                    "plain" in model_name.lower()
-                    and "finetune" not in model_name.lower()
-                ):
-                    template_name = "v1_mmtag"
-                else:
-                    template_name = "llava_v1"
-            elif "mpt" in model_name.lower():
-                template_name = "mpt"
-            else:
-                if "mmtag" in model_name.lower():
-                    template_name = "v0_mmtag"
-                elif (
-                    "plain" in model_name.lower()
-                    and "finetune" not in model_name.lower()
-                ):
-                    template_name = "v0_mmtag"
-                else:
-                    template_name = "llava_v0"
-        elif "mpt" in model_name:
-            template_name = "mpt_text"
-        elif "llama-2" in model_name:
-            template_name = "llama_2"
-        else:
-            template_name = "vicuna_v1"
+        template_name = 'phi'
+        # if "tinyllava" in model_name.lower():
+        #     if "3.1b" in model_name.lower() or "phi" in model_name.lower():
+        #         template_name = "phi"
+        #     elif "2.0b" in model_name.lower() or "stablelm" in model_name.lower():
+        #         template_name = "phi"
+        #     elif "qwen" in model_name.lower():
+        #         template_name = "qwen"
+        #     else:
+        #         template_name = "v1"
+
+        # elif "llava" in model_name.lower():
+
+        #     if "llama-2" in model_name.lower():
+        #         template_name = "llava_llama_2"
+        #     elif "v1" in model_name.lower():
+        #         if "mmtag" in model_name.lower():
+        #             template_name = "v1_mmtag"
+        #         elif (
+        #             "plain" in model_name.lower()
+        #             and "finetune" not in model_name.lower()
+        #         ):
+        #             template_name = "v1_mmtag"
+        #         else:
+        #             template_name = "llava_v1"
+        #     elif "mpt" in model_name.lower():
+        #         template_name = "mpt"
+        #     else:
+        #         if "mmtag" in model_name.lower():
+        #             template_name = "v0_mmtag"
+        #         elif (
+        #             "plain" in model_name.lower()
+        #             and "finetune" not in model_name.lower()
+        #         ):
+        #             template_name = "v0_mmtag"
+        #         else:
+        #             template_name = "llava_v0"
+        # elif "mpt" in model_name:
+        #     template_name = "mpt_text"
+        # elif "llama-2" in model_name:
+        #     template_name = "llama_2"
+        # else:
+        #     template_name = "vicuna_v1"
         new_state = conv_templates[template_name].copy()
         new_state.append_message(new_state.roles[0], state.messages[-2][1])
         new_state.append_message(new_state.roles[1], None)