upgate server

model/model_config.py

@@ -26,6 +26,12 @@ register_model_config(
 #     online_model=False
 # )
 
+register_model_config(
+    model_name="grm-t",
+    i2s_model=False,
+    online_model=False
+)
+
 register_model_config(
     model_name="latent-nerf",
     i2s_model=False,
@@ -50,7 +56,6 @@ register_model_config(
     online_model=False
 )
 
-
 register_model_config(
     model_name="point-e-t",
     i2s_model=False,
@@ -68,11 +73,6 @@ register_model_config(
     i2s_model=False,
     online_model=False
 )
-# register_model_config(
-#     model_name="prolificdreamer",
-#     i2s_model=False,
-#     online_model=False
-# )
 
 # register_model_config(
 #     model_name="dreamgaussian",
@@ -106,11 +106,16 @@ register_model_config(
 )
 
 register_model_config(
-    model_name="openlrm",
+    model_name="grm-i",
     i2s_model=True,
     online_model=False
 )
 
+register_model_config(
+    model_name="openlrm",
+    i2s_model=True,
+    online_model=False
+)
 
 register_model_config(
     model_name="point-e-i",
@@ -125,13 +130,13 @@ register_model_config(
 )
 
 register_model_config(
-    model_name="syncdreamer",
+    model_name="stable-zero123",
     i2s_model=True,
     online_model=False
 )
 
 register_model_config(
-    model_name="wonder3d",
+    model_name="syncdreamer",
     i2s_model=True,
     online_model=False
 )
@@ -143,13 +148,13 @@ register_model_config(
 )
 
 register_model_config(
-    model_name="zero123-xl",
+    model_name="wonder3d",
     i2s_model=True,
     online_model=False
 )
 
 register_model_config(
-    model_name="stable-zero123",
+    model_name="zero123-xl",
     i2s_model=True,
     online_model=False
-)
+)

model/model_registry.py

@@ -102,6 +102,14 @@ register_model_info(
     "Multi-view Diffusion for 3D Generation",
 )
 
+
+register_model_info(
+    ["grm-t", "grm-i"],
+    "GRM",
+    "https://justimyhxu.github.io/projects/grm",
+    "GRM: Large Gaussian Reconstruction Model for Efficient 3D Reconstruction and Generation",
+)
+
 register_model_info(
     ["point-e-t", "point-e-i"],
     "Point·E",

serve/vote_utils.py

@@ -7,12 +7,19 @@ from pathlib import Path
 from .utils import *
 from .log_utils import build_logger
 from constants import IMAGE_DIR, EVALUATE_DIMS
+from model.model_registry import model_info
 
 t2s_logger = build_logger("gradio_web_server_text2shape", "gr_web_text2shape.log") # t2s = image generation, loggers for single model direct chat
 t2s_multi_logger = build_logger("gradio_web_server_text2shape_multi", "gr_web_text2shape_multi.log") # t2s_multi = image generation multi, loggers for side-by-side and battle
 i2s_logger = build_logger("gradio_web_server_image2shape", "gr_web_image2shape.log") # i2s = image editing, loggers for single model direct chat
 i2s_multi_logger = build_logger("gradio_web_server_image2shape_multi", "gr_web_image2shape_multi.log") # i2s_multi = image editing multi, loggers for side-by-side and battle
 
+def remove_suffix(name):
+    if name.endswith("-t") or name.endswith("-i"):
+        return name[:-2]
+    else:
+        return name
+
 def vote_last_response_t2s(state, dim, vote_type, model_selector, request: gr.Request):
     with open(get_conv_log_filename(), "a") as fout:
         data = {
@@ -178,8 +185,8 @@ def leftvote_last_response_t2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
 
 def rightvote_last_response_t2s_anony(
     state0, state1, model_selector0, model_selector1, dim_md, request: gr.Request
@@ -199,8 +206,8 @@ def rightvote_last_response_t2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
 
 def tievote_last_response_t2s_anony(
     state0, state1, model_selector0, model_selector1, dim_md, request: gr.Request
@@ -220,8 +227,8 @@ def tievote_last_response_t2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
     
 def bothbad_vote_last_response_t2s_anony(
     state0, state1, model_selector0, model_selector1, dim_md, request: gr.Request
@@ -241,8 +248,8 @@ def bothbad_vote_last_response_t2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
 
 ## Image-to-Shape (i2s) Single Model Direct Chat
 def upvote_last_response_i2s(state, model_selector, dim_md, request: gr.Request):
@@ -331,8 +338,8 @@ def leftvote_last_response_i2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
         
 
 def rightvote_last_response_i2s_anony(
@@ -353,8 +360,8 @@ def rightvote_last_response_i2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
         
 
 def tievote_last_response_i2s_anony(
@@ -375,8 +382,8 @@ def tievote_last_response_i2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))
         
 
 def bothbad_vote_last_response_i2s_anony(
@@ -397,8 +404,8 @@ def bothbad_vote_last_response_i2s_anony(
     is_visible = (state0.evaluted_dims == state1.evaluted_dims == EVALUATE_DIMS)
     return (state0, state1) \
         + (disable_btn,) * 4 \
-        + (gr.Markdown(f"### Model A: {state0.model_name}", visible=is_visible), 
-           gr.Markdown(f"### Model B: {state1.model_name}", visible=is_visible))
+        + (gr.Markdown(f"### Model A: {model_info[state0.model_name].simple_name}", visible=is_visible), 
+           gr.Markdown(f"### Model B: {model_info[state1.model_name].simple_name}", visible=is_visible))