fix rerun

serve/constants.py

@@ -17,16 +17,4 @@ SAVE_IMAGE = "save_image"
 SAVE_VIDEO = "save_video"
 SAVE_LOG = "save_log"
 
-IMAGE_GENERATION_MODELS = ['fal_LCM(v1.5/XL)_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
-                            'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
-                            'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
-IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
-                        'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
-                        'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
-                        'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
-VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
-                           'fal_AnimateDiffTurbo_text2video',
-                           'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
-                           'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
-
 

serve/vote_utils.py

@@ -7,7 +7,7 @@ import regex as re
 from pathlib import Path
 from .utils import *
 from .log_utils import build_logger
-from .constants import IMAGE_DIR, VIDEO_DIR, IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, VIDEO_GENERATION_MODELS
+from .constants import IMAGE_DIR, VIDEO_DIR
 import imageio
 
 ig_logger = build_logger("gradio_web_server_image_generation", "gr_web_image_generation.log") # ig = image generation, loggers for single model direct chat
@@ -598,13 +598,8 @@ def generate_igm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
     ip = get_ip(request)
     igm_logger.info(f"generate. ip: {ip}")
     start_tstamp = time.time()
-    model_name0 = re.sub(r"### Model A: ", "", model_name0)
-    model_name1 = re.sub(r"### Model B: ", "", model_name1)
-    model_map = {model_name.split('_')[1]: model_name for model_name in IMAGE_GENERATION_MODELS}
-    if model_name0 in model_map:
-        model_name0 = model_map[model_name0]
-    if model_name1 in model_map:
-        model_name1 = model_map[model_name1]
+    model_name0 = ""
+    model_name1 = ""
     generated_image0, generated_image1, model_name0, model_name1 = gen_func(text, model_name0, model_name1)
     state0.prompt = text
     state1.prompt = text
@@ -801,13 +796,8 @@ def generate_iem_annoy(gen_func, state0, state1, source_text, target_text, instr
     ip = get_ip(request)
     igm_logger.info(f"generate. ip: {ip}")
     start_tstamp = time.time()
-    model_name0 = re.sub(r"### Model A: ", "", model_name0)
-    model_name1 = re.sub(r"### Model B: ", "", model_name1)
-    model_map = {model_name.split('_')[1]: model_name for model_name in IMAGE_EDITION_MODELS}
-    if model_name0 in model_map:
-        model_name0 = model_map[model_name0]
-    if model_name1 in model_map:
-        model_name1 = model_map[model_name1]
+    model_name0 = ""
+    model_name1 = ""
     generated_image0, generated_image1, model_name0, model_name1 = gen_func(source_text, target_text, instruct_text, source_image, model_name0, model_name1)
     state0.source_prompt = source_text
     state0.target_prompt = target_text
@@ -1000,13 +990,8 @@ def generate_vgm_annoy(gen_func, state0, state1, text, model_name0, model_name1,
     ip = get_ip(request)
     vgm_logger.info(f"generate. ip: {ip}")
     start_tstamp = time.time()
-    model_name0 = re.sub(r"### Model A: ", "", model_name0)
-    model_name1 = re.sub(r"### Model B: ", "", model_name1)
-    model_map = {model_name.split('_')[1]: model_name for model_name in VIDEO_GENERATION_MODELS}
-    if model_name0 in model_map:
-        model_name0 = model_map[model_name0]
-    if model_name1 in model_map:
-        model_name1 = model_map[model_name1]
+    model_name0 = ""
+    model_name1 = ""
     generated_video0, generated_video1, model_name0, model_name1 = gen_func(text, model_name0, model_name1)
     state0.prompt = text
     state1.prompt = text