fix

serve/__init__.py

@@ -5,31 +5,24 @@ from collections import OrderedDict
 MODEL_ID_TO_NAME = OrderedDict(
     [
         (
-            "llava-lvis4v-lrv+redux-lvis4v-lrv-resize-naive-dinosiglip-vit-so-14-384px-no-align+13b+stage-finetune+x7",
-            "PrismaticVLM 13B - Chat",
-        ),
-        (
-            "llava-lvis4v-lrv+redux-lvis4v-lrv-resize-naive-dinosiglip-vit-so-14-384px-no-align+7b+stage-finetune+x7",
-            "PrismaticVLM 7B - Chat",
-        ),
-        (
-            "llava-lvis4v-lrv+redux-lvis4v-lrv-resize-naive-dinosiglip-vit-so-14-384px-no-align-llama2pure+13b+stage-finetune+x7",
+            "prism-dinosiglip+13b",
             "PrismaticVLM 13B",
         ),
         (
-            "llava-lvis4v-lrv+redux-lvis4v-lrv-resize-naive-dinosiglip-vit-so-14-384px-no-align-llama2pure+7b+stage-finetune+x7",
+            "prism-dinosiglip+7b",
             "PrismaticVLM 7B",
         ),
         (
-            "redux-resize-naive-dinosiglip-vit-so-14-384px-no-align-llama2pure+13b+stage-finetune+x7",
+            "prism-dinosiglip-controlled+13b",
             "PrismaticVLM 13B (Controlled)",
         ),
         (
-            "redux-resize-naive-dinosiglip-vit-so-14-384px-no-align-llama2pure+7b+stage-finetune+x7",
+            "prism-dinosiglip-controlled+7b",
             "PrismaticVLM 7B (Controlled)",
         ),
-        ("llava-v1.5-13b", "LLaVA 1.5: 13B"),
-        ("llava-v1.5-7b", "LLaVA 1.5: 7B"),
+        ("llava-v1.5-13b", "LLaVA 1.5 13B"),
+        ("llava-v1.5-7b", "LLaVA 1.5 7B"),
+        ("instructblip-vicuna-7b", "InstructBLIP 7B"),
     ]
 )
 

serve/gradio_web_server.py

@@ -93,24 +93,6 @@ def vote_last_response(state, vote_type, model_selector, request: gr.Request):
         fout.write(json.dumps(data) + "\n")
 
 
-# def upvote_last_response(state, model_selector, request: gr.Request):
-#     logger.info(f"upvote. ip: {request.client.host}")
-#     vote_last_response(state, "upvote", model_selector, request)
-#     return ("",) + (disable_btn,) * 3
-
-
-# def downvote_last_response(state, model_selector, request: gr.Request):
-#     logger.info(f"downvote. ip: {request.client.host}")
-#     vote_last_response(state, "downvote", model_selector, request)
-#     return ("",) + (disable_btn,) * 3
-
-
-# def flag_last_response(state, model_selector, request: gr.Request):
-#     logger.info(f"flag. ip: {request.client.host}")
-#     vote_last_response(state, "flag", model_selector, request)
-#     return ("",) + (disable_btn,) * 3
-
-
 def regenerate(state, image_process_mode, request: gr.Request):
     logger.info(f"regenerate. ip: {request.client.host}")
     state.messages[-1][-1] = None
@@ -388,15 +370,6 @@ def build_demo(embed_mode):
 
         # Register listeners
         btn_list = [regenerate_btn, clear_btn]
-        # upvote_btn.click(
-        #     upvote_last_response, [state, model_selector], [textbox, upvote_btn, downvote_btn, flag_btn], queue=False
-        # )
-        # downvote_btn.click(
-        #     downvote_last_response, [state, model_selector], [textbox, upvote_btn, downvote_btn, flag_btn], queue=False
-        # )
-        # flag_btn.click(
-        #     flag_last_response, [state, model_selector], [textbox, upvote_btn, downvote_btn, flag_btn], queue=False
-        # )
 
         regenerate_btn.click(
             regenerate, [state, image_process_mode], [state, chatbot, textbox, imagebox, *btn_list], queue=False