Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
Commit
•
e049190
1
Parent(s):
1d20838
add LCM XL
Browse files- model/model_registry.py +7 -0
- model/models/__init__.py +1 -1
- model/models/fal_api_models.py +1 -1
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 = ['
|
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):
|