tianleliphoebe commited on
Commit
e049190
1 Parent(s): 1d20838

add LCM XL

Browse files
model/model_registry.py CHANGED
@@ -52,6 +52,13 @@ register_model_info(
52
  "Latent Consistency Models.",
53
  )
54
 
 
 
 
 
 
 
 
55
  register_model_info(
56
  ["imagenhub_PlayGroundV2_generation", 'playground_PlayGroundV2_generation'],
57
  "Playground v2",
 
52
  "Latent Consistency Models.",
53
  )
54
 
55
+ register_model_info(
56
+ ["fal_LCM(v1.5/XL)_text2image"],
57
+ "LCM(v1.5/XL)",
58
+ "https://fal.ai/models/fast-lcm-diffusion-turbo",
59
+ "Latent Consistency Models (v1.5/XL)",
60
+ )
61
+
62
  register_model_info(
63
  ["imagenhub_PlayGroundV2_generation", 'playground_PlayGroundV2_generation'],
64
  "Playground v2",
model/models/__init__.py CHANGED
@@ -4,7 +4,7 @@ from .fal_api_models import load_fal_model
4
  from .videogenhub_models import load_videogenhub_model
5
 
6
 
7
- IMAGE_GENERATION_MODELS = ['fal_LCM_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
8
  'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
9
  'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
10
  IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
 
4
  from .videogenhub_models import load_videogenhub_model
5
 
6
 
7
+ IMAGE_GENERATION_MODELS = ['fal_LCM(v1.5/XL)_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
8
  'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
9
  'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
10
  IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
model/models/fal_api_models.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import base64
7
 
8
  FAL_MODEl_NAME_MAP = {"SDXL": "fast-sdxl", "SDXLTurbo": "fast-turbo-diffusion", "SDXLLightning": "fast-lightning-sdxl",
9
- "LCM": "fast-lcm-diffusion", "PixArtSigma": "pixart-sigma", "StableCascade": "stable-cascade"}
10
 
11
  class FalModel():
12
  def __init__(self, model_name, model_type):
 
6
  import base64
7
 
8
  FAL_MODEl_NAME_MAP = {"SDXL": "fast-sdxl", "SDXLTurbo": "fast-turbo-diffusion", "SDXLLightning": "fast-lightning-sdxl",
9
+ "LCM(v1.5/XL)": "fast-lcm-diffusion", "PixArtSigma": "pixart-sigma", "StableCascade": "stable-cascade"}
10
 
11
  class FalModel():
12
  def __init__(self, model_name, model_type):