monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Introduce qwen-image-layered model saving

Browse Source
This commit is contained in:
Zimeng Xiong 2025-12-22 08:21:24 -08:00
parent fe6936449d
commit 067ba25c7e
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/mflux/cli/defaults/defaults.py
View File

@ -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,
}

5
src/mflux/models/common/cli/save.py
View File

@ -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