diff --git a/src/mflux/cli/defaults/defaults.py b/src/mflux/cli/defaults/defaults.py index 4aaadef..1f4c20d 100644 --- a/src/mflux/cli/defaults/defaults.py +++ b/src/mflux/cli/defaults/defaults.py @@ -12,12 +12,13 @@ GUIDANCE_SCALE = 3.5 GUIDANCE_SCALE_KONTEXT = 2.5 HEIGHT, WIDTH = 1024, 1024 IMAGE_STRENGTH = 0.4 -MODEL_CHOICES = ["dev", "schnell", "krea-dev", "dev-krea", "qwen", "fibo", "z-image-turbo"] +MODEL_CHOICES = ["dev", "schnell", "krea-dev", "dev-krea", "qwen", "qwen-image-layered", "fibo", "z-image-turbo"] MODEL_INFERENCE_STEPS = { "dev": 25, "schnell": 4, "krea-dev": 25, "qwen": 20, + "qwen-image-layered": 50, "fibo": 20, "z-image-turbo": 9, } diff --git a/src/mflux/models/common/cli/save.py b/src/mflux/models/common/cli/save.py index 7751313..99ceb16 100644 --- a/src/mflux/models/common/cli/save.py +++ b/src/mflux/models/common/cli/save.py @@ -3,6 +3,7 @@ from mflux.models.common.config import ModelConfig from mflux.models.fibo.variants.txt2img.fibo import FIBO from mflux.models.flux.variants.txt2img.flux import Flux1 from mflux.models.qwen.variants.txt2img.qwen_image import QwenImage +from mflux.models.qwen_layered.variants.i2l.qwen_image_layered import QwenImageLayered from mflux.models.z_image.variants.turbo.z_image_turbo import ZImageTurbo @@ -15,7 +16,9 @@ def main(): # 1. Determine model class based on model name model_name_lower = args.model.lower() - if "qwen" in model_name_lower: + if "qwen" in model_name_lower and "layered" in model_name_lower: + model_class = QwenImageLayered + elif "qwen" in model_name_lower: model_class = QwenImage elif "fibo" in model_name_lower: model_class = FIBO