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)