Add Qwen-Image-Distill-Full

This commit is contained in:
Disty0
2025-09-06 19:24:15 +03:00
parent 65b6d35429
commit 08477b99d4
3 changed files with 18 additions and 3 deletions
+9 -2
View File
@@ -192,17 +192,24 @@
"Qwen-Image-Edit": {
"path": "Qwen/Qwen-Image-Edit",
"preview": "Qwen--Qwen-Image-Edit.jpg",
"desc": " Qwen-Image-Edit, the image editing version of Qwen-Image. Built upon our 20B Qwen-Image model, Qwen-Image-Edit successfully extends Qwen-Images unique text rendering capabilities to image editing tasks, enabling precise text editing.",
"desc": "Qwen-Image-Edit, the image editing version of Qwen-Image. Built upon our 20B Qwen-Image model, Qwen-Image-Edit successfully extends Qwen-Images unique text rendering capabilities to image editing tasks, enabling precise text editing.",
"skip": true,
"extras": ""
},
"Qwen-Image-Lightning": {
"path": "vladmandic/Qwen-Lightning",
"preview": "Qwen-Lightning.jpg",
"desc": " Qwen-Lightning is step-distilled from Qwen-Image to allow for generation in 8 steps.",
"desc": "Qwen-Lightning is step-distilled from Qwen-Image to allow for generation in 8 steps.",
"skip": true,
"extras": "steps: 8"
},
"Qwen-Image-Distill-Full": {
"path": "SahilCarterr/Qwen-Image-Distill-Full",
"preview": "Qwen-Image-Distill-Full.jpg",
"desc": "Qwen-Image-Distill-Full is a distilled and accelerated version of Qwen-Image by DiffSynth-Studio.",
"skip": true,
"extras": "steps: 15"
},
"Ostris Flex.2 Preview": {
"path": "ostris/Flex.2-preview",
Binary file not shown.

After

Width:  |  Height:  |  Size: 75 KiB

+9 -1
View File
@@ -29,8 +29,16 @@ def load_qwen(checkpoint_info, diffusers_load_config={}):
# if transformer is not None:
# cls_name = nunchaku.pipeline.pipeline_qwenimage.NunchakuQwenImagePipeline # we dont need this
if 'Qwen-Image-Distill-Full' in repo_id:
repo_transformer = repo_id
transformer_subfolder = None
repo_id = 'Qwen/Qwen-Image'
else:
repo_transformer = repo_id
transformer_subfolder = "transformer"
if transformer is None:
transformer = generic.load_transformer(repo_id, cls_name=diffusers.QwenImageTransformer2DModel, load_config=diffusers_load_config, modules_dtype_dict={"minimum_6bit": ["pos_embed", "time_text_embed", "img_in", "txt_in", "norm_out", "transformer_blocks.0.img_mod.1.weight"]})
transformer = generic.load_transformer(repo_transformer, subfolder=transformer_subfolder, cls_name=diffusers.QwenImageTransformer2DModel, load_config=diffusers_load_config, modules_dtype_dict={"minimum_6bit": ["pos_embed", "time_text_embed", "img_in", "txt_in", "norm_out", "transformer_blocks.0.img_mod.1.weight"]})
repo_te = 'Qwen/Qwen-Image' # if 'Qwen-Lightning' in repo_id or 'Qwen-Image-Edit' in repo_id else repo_id
text_encoder = generic.load_text_encoder(repo_te, cls_name=transformers.Qwen2_5_VLForConditionalGeneration, load_config=diffusers_load_config)