diff --git a/html/reference.json b/html/reference.json
index ae8c37e85..091fe5e80 100644
--- a/html/reference.json
+++ b/html/reference.json
@@ -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-Image’s 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-Image’s 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",
diff --git a/models/Reference/Qwen-Image-Distill-Full.jpg b/models/Reference/Qwen-Image-Distill-Full.jpg
new file mode 100644
index 000000000..ac05ebcee
Binary files /dev/null and b/models/Reference/Qwen-Image-Distill-Full.jpg differ
diff --git a/pipelines/model_qwen.py b/pipelines/model_qwen.py
index 2af23595c..de867ca8c 100644
--- a/pipelines/model_qwen.py
+++ b/pipelines/model_qwen.py
@@ -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)