renaming FLUX

model/fetch_museum_results/__init__.py

@@ -2,7 +2,7 @@ from .imagen_museum import TASK_DICT, DOMAIN
 from .imagen_museum import fetch_indexes, fetch_indexes_no_csv
 import random
 
-ARENA_TO_IG_MUSEUM = {"LCM(v1.5/XL)":"LCM", "PlayGroundV2.5": "PlayGroundV2_5"}
+ARENA_TO_IG_MUSEUM = {"LCM(v1.5/XL)":"LCM", "PlayGroundV2.5": "PlayGroundV2_5", "FLUX.1-schnell": "FLUX1schnell", "FLUX.1-dev": "FLUX1dev"}
 ARENA_TO_VG_MUSEUM = {"StableVideoDiffusion": "FastSVD"}
 
 def draw2_from_imagen_museum(task, model_name1, model_name2):

model/model_registry.py

@@ -151,14 +151,14 @@ register_model_info(
 )
 
 register_model_info(
-    ["fal_FluxTimestep_text2image"],
+    ["fal_FLUX1schnell_text2image"],
     "FLUX.1-schnell",
     "https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux",
     "Flux is a series of text-to-image generation models based on diffusion transformers. Timestep-distilled version.",
 )
 
 register_model_info(
-    ["fal_FluxGuidance_text2image"],
+    ["fal_FLUX1dev_text2image"],
     "FLUX.1-dev",
     "https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux",
     "Flux is a series of text-to-image generation models based on diffusion transformers. Guidance-distilled version.",

model/models/__init__.py

@@ -10,7 +10,7 @@ from .videogenhub_models import load_videogenhub_model
 IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
                            'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation', 'imagenhub_HunyuanDiT_generation',
                            'playground_PlayGroundV2.5_generation', 'imagenhub_Kolors_generation', 'imagenhub_SD3_generation',
-                           'fal_AuraFlow_text2image', 'fal_FluxTimestep_text2image', 'fal_FluxGuidance_text2image']  # 'playground_PlayGroundV2_generation'
+                           'fal_AuraFlow_text2image', 'fal_FLUX1schnell_text2image', 'fal_FLUX1dev_text2image']  # 'playground_PlayGroundV2_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', 

model/models/fal_api_models.py

@@ -7,7 +7,7 @@ import base64
 
 FAL_MODEl_NAME_MAP = {"SDXL": "fast-sdxl", "SDXLTurbo": "fast-turbo-diffusion", "SDXLLightning": "fast-lightning-sdxl",
                       "LCM(v1.5/XL)": "fast-lcm-diffusion", "PixArtSigma": "pixart-sigma", "StableCascade": "stable-cascade",
-                      "AuraFlow": "aura-flow", "FluxTimestep": "flux/schnell", "FluxGuidance": "flux/dev"}
+                      "AuraFlow": "aura-flow", "FLUX1schnell": "flux/schnell", "FLUX1dev": "flux/dev"}
 
 class FalModel():
     def __init__(self, model_name, model_type):