fix: update naming and align with repo

Co-authored-by: Cursor <cursoragent@cursor.com>

.gitattributes

@@ -43,3 +43,4 @@ assets/Pruna-Q-Image-2.1-t2i-2048.png filter=lfs diff=lfs merge=lfs -text
 assets/Pruna-Q-Image-2.1.png filter=lfs diff=lfs merge=lfs -text
 assets/*.png filter=lfs diff=lfs merge=lfs -text
 assets/Pruna-Qwen-Image-2.1.png filter=lfs diff=lfs merge=lfs -text
+P-Q-Image-2.1.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,7 +1,7 @@
 ---
 license: other
 license_name: qwen-research
-base_model: PrunaAI/Pruna-Q-Image-2.1
+base_model: PrunaAI/P-Q-Image-2.1
 pipeline_tag: text-to-image
 tags:
   - comfyui
@@ -12,11 +12,59 @@ tags:
   - distilled
   - few-step
 ---
-
-# Pruna-Q-Image-2.1 for ComfyUI
-
-ComfyUI-ready copies of the [Pruna-Q-Image-2.1](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1)
-few-step LoRA adapters for [Qwen-Image-2.1](https://huggingface.co/Qwen/Qwen-Image-2.1),
+<!-- header start -->
+<!-- 200823 -->
+<p align="center">
+  <img src="P-Q-Image-2.1.png" width="800">
+</p>
+<!-- header end -->
+
+<p align="center">
+  <a href="https://github.com/PrunaAI/pruna">
+    <img src="https://img.shields.io/badge/GitHub-PrunaAI-9334E9?style=plastic&logo=github&logoColor=white" alt="GitHub">
+  </a>
+  &nbsp;
+  <a href="https://twitter.com/PrunaAI">
+    <img src="https://img.shields.io/badge/Twitter%2FX-@PrunaAI-9334E9?style=plastic&logo=x&logoColor=white" alt="Twitter/X">
+  </a>
+  &nbsp;
+  <a href="https://www.linkedin.com/company/pruna-ai">
+    <img src="https://img.shields.io/badge/LinkedIn-PrunaAI-9334E9?style=plastic&logo=linkedin&logoColor=white" alt="LinkedIn">
+  </a>
+  &nbsp;
+  <a href="https://discord.com/invite/JFQmtFKCjd">
+    <img src="https://img.shields.io/badge/Discord-Join%20us-9334E9?style=plastic&logo=discord&logoColor=white" alt="Discord">
+  </a>
+  &nbsp;
+  <a href="https://dashboard.pruna.ai/login?utm_source=huggingface&utm_medium=org_card&utm_campaign=hf_traffic">
+    <img src="https://img.shields.io/badge/Performance%20Models-Try%20them%20now-9334E9?style=plastic&logo=data:image/svg%2bxml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHdpZHRoPSI0MTEiIGhlaWdodD0iNDc3IiBmaWxsPSJub25lIj48cGF0aCBmaWxsPSIjZmZmIiBkPSJNMjMuNzAzIDMwMC41MzJjLTEuODQyLS42NzQtNC4xNDItMS4wNTctNS43MDMtMi4wMjktMTIuNTU4LTcuODQ2LTguODM3LTI1LjE3MiA1LjI1NS0yOC4zNjFhLjY2LjY2IDAgMCAwIC4zNDgtLjIwMS42LjYgMCAwIDAgLjE1Ni0uMzZxNi40OS04MS45ODYgNjguNTc3LTEzNC42MjlhLjM0LjM0IDAgMCAwIC4wOTItLjEyNS4zMy4zMyAwIDAgMC0uMDI2LS4zLjMzLjMzIDAgMCAwLS4xMTItLjEwOGMtNC4xMTUtMi40OTctOC42NC00Ljc1LTEyLjM1My03LjQ0M2ExOTAgMTkwIDAgMCAxLTE4Ljc5NC0xNS41MjJjLTIuOTM2LTIuNzY4LTMuNzAzLTUuODkxLjU4LTcuNzdhLjUuNSAwIDAgMCAuMjgtLjYwOEM1MyA3NS45NSA0OC40MTcgNDguNTk5IDQ1LjUxOCAxOS42NXEtLjM5My0zLjg1Mi0uMjcxLTYuMzJhLjc1Ljc1IDAgMCAxIC4xMzQtLjQuNy43IDAgMCAxIC4zMjQtLjI1NXEyLjEwNC0uODQyIDUuODQ0LTEuNDMxYzU4LjExNC05LjEzNiAxMjcuMDA5IDguNDI1IDE0NC4zMDcgNzIuNjQ1cS4xMDMuMzY0LjE3OCAwIDUuMDk2LTI0LjA5NyA0LjYzOC00OC4xOTMtLjA0Ny0yLjc4NyAxLjg3OS00Ljc4OCAxMC40MDctMTAuOCAyNC43OTgtOC42ODdjNy40NzEgMS4wOTUgNy43NDIgNC42MzggNy4yMDkgMTEuMWEzMzQgMzM0IDAgMCAxLTExLjg4NCA2NC43MDYuMzg4LjM4OCAwIDAgMCAuMTgyLjQ0NC40LjQgMCAwIDAgLjE2NC4wNTFjOTIuNDIxIDguNDUzIDE1OC44NzUgNzkuNDg5IDE2NS4wODQgMTcxLjI3NGExIDEgMCAwIDAgLjEyOS40MjQuODMuODMgMCAwIDAgLjI5MS4yOTZjMy43NSAyLjE3IDYuMiAyLjIxNiA4Ljg1NSA2LjU5MiA1LjA2OCA4LjMzMiAxLjQ1IDIwLjQzMS04LjU5MyAyMi40N2ExLjcxIDEuNzEgMCAwIDAtMS4zNjUgMS40NjhxLTkuMTczIDc1LjM4My02NC43NjIgMTIzLjM3MWMtMjguNjIyIDI0LjcwNC02NS43NDQgMzkuNzQ5LTEwMy40NjQgNDIuMzMtMTAwLjM1OSA2Ljg2My0xODQuNDg2LTY1LjYyMi0xOTUuMDA1LTE2NS41N2EuOC44IDAgMCAwLS4xNTQtLjM5NS43NC43NCAwIDAgMC0uMzMzLS4yNU0xNTEuMzM4IDk3LjY0M2wxOS4wNjYgMTkuMzc1YS41LjUgMCAwIDAgLjE1LjEuNS41IDAgMCAwIC4xNzYuMDMxLjQzLjQzIDAgMCAwIC4zMTktLjE1cTEwLjg0Ny0xMi40NjUgNy40NDMtMjguMTU1Yy00LjUzNS0yMC44Ni0xOC41OTgtMzcuNjY0LTM3LjI3MS00OC4zODktMjEuMDY3LTEyLjY4OS00Ny44NjUtMTUuOTA1LTcyLjI4LTE0Ljg0cS0uNjE3LjAzLS4xNC40MjJsNjMuNDcyIDUzLjMyNmEuNC40IDAgMCAwIC4xNDQuMDcyLjM0LjM0IDAgMCAwIC4zMDEtLjA2OC4zLjMgMCAwIDAgLjA4OC0uMTI2IDQ3IDQ3IDAgMCAwIDMuMjgyLTEzLjA4MXEuMDIzLS4yMzMuMTI0LS40MzN0LjI2OC0uMzM0YzIuOTU1LTIuNDc4IDQuMTE1LjQ1OCA0LjY5NCAyLjg1MnEyLjMyIDkuNTM3LjE4NyAxOS4yNDRhMS40OCAxLjQ4IDAgMCAwIC40NCAxLjQwM3ptLTYuNTI2IDkuMjk1LTkuMDI0LTEwLjI3NmEuNy43IDAgMCAwLS4zNDMtLjIwNy44Ni44NiAwIDAgMC0uNDMzIDBxLTExLjY4IDMuMDIxLTIzLjExNC0uODIyYy01LjgxNi0zLjg1Mi0uNzU4LTUuNTE3IDIuNTktNS44MTZxNS45MTgtLjUzMyAxMS43MzUtMi41MDZhLjMuMyAwIDAgMCAuMTI3LS4wOC4zLjMgMCAwIDAgLjA3My0uMTMzLjM0LjM0IDAgMCAwIC4wMDEtLjE1NC4zNC4zNCAwIDAgMC0uMDctLjEzOGMtMTkuNDU5LTIxLjE1LTQwLjc2LTQwLjI0NS02Mi4yNzUtNTkuMjY0cS0uMzU2LS4zMTgtLjMxOC4xNWEzMDggMzA4IDAgMCAwIDguMzg3IDUxLjgwMmM0LjU5MiAxOC4zMDggMTMuNzU1IDM0LjE1NyAzMS44NDggNDEuODcycTI2LjI0NyAxMS4xODIgNTMuNzI5IDIuODMzYS41ODMuNTgzIDAgMCAwIC4yOTktLjg4OCAyMzUgMjM1IDAgMCAwLTEzLjIxMi0xNi4zNzNtNDUuMTI1IDQ5Ljc0NXEtMTIuNTk1IDEuMzE4LTIzLjQ0MS01LjU4M2MtMy4wNzctNC44OTkgMi4wNjYtOC41OTMgNS40NDItMTEuMTgzYS41ODQuNTg0IDAgMCAwIC4xODctLjcwMSA4LjQ2IDguNDYgMCAwIDAtMy40ODgtMy44NjIuODEuODEgMCAwIDAtLjgyMy4wMDljLTE2LjY5MSA5Ljk3Ny0zNC40ODUgMTAuNzgyLTUzLjE4NiA2Ljk2N3EtMS41Ny0uMzItMi44NTIuNjQ1LTYxLjQ1MSA0Ni4zNy02OS42NyAxMjMuOTEzLS42ODMgNi40MzMtMS44MDYgOC44MjdjLTIuOTQ1IDYuMzAyLTcuNzcgNy44NDUtMTQuMjIyIDkuNDcycS0uNDQ5LjExMyAwIC4yMzRjOS4yNzYgMi4zMSAxNC42OCA2LjE0MyAxNi4wNzQgMTYuNDAxIDguNDYyIDYyLjQ5IDQ4Ljk0IDExNy44NjQgMTEwLjM4MyAxMzguMzQxYTEwNCAxMDQgMCAwIDAgMTEuMDA2IDMuNTA3YzkyLjY0NSAyMy45MDkgMTgxLjIzMi0zMy42NjIgMjAyLjgwNC0xMjUuMDU0cTEuNjA5LTYuNzkgMy40NS0xOS4wODVjMS4xMDQtNy4zOTYgNS42OTUtMTIuMTE4IDEzLjE0Ny0xMy41NjguNDEyLS4wODQgMS45MDgtLjc1Ny4yODEtLjk5MS04LjA1MS0xLjE0MS0xMi4yNC02LjM4Ni0xMi45ODgtMTQuMTI4cS0xLjA3NS0xMS4yNS0yLjUxNi0xOS4wMzhjLTExLjc3Mi02My44OTItNTYuMzgzLTExNC41NzItMTE5LjI4NS0xMzEuNDZxLTE1LjEyLTMuODgtMzAuOTg3LTUuMjQ1YS4zODYuMzg2IDAgMCAwLS40MDIuMjUyIDE2OC42IDE2OC42IDAgMCAxLTExLjM1MiAyNC44NTRjLTQuMTMzIDcuNDA2LTguMTYzIDE1LjQ5NC0xNS43NTYgMTYuNDc2Ii8+PGNpcmNsZSBjeD0iMTQwLjA2NSIgY3k9IjI3My4wMyIgcj0iNTAuNjE4IiBzdHJva2U9IiNmZmYiIHN0cm9rZS13aWR0aD0iOS4yMDMiLz48cGF0aCBmaWxsPSIjZmZmIiBkPSJNMTQ2Ljk2NyAyNDIuMDkyYzE1LjY3MiAwIDI4LjM3NiAxMy45NjMgMjguMzc2IDMxLjE4OHMtMTIuNzA0IDMxLjE4OS0yOC4zNzYgMzEuMTg5LTI4LjM3Ny0xMy45NjQtMjguMzc3LTMxLjE4OWMwLTIuNjI1LjI5Ny01LjE3NC44NTItNy42MSAxLjYwNyAzLjcyNyA1LjMxMyA2LjMzOCA5LjYyOSA2LjMzOCA1Ljc4OSAwIDEwLjQ4MS00LjY5MyAxMC40ODItMTAuNDgycy00LjY5My0xMC40ODEtMTAuNDgyLTEwLjQ4MWMtLjc2MSAwLTEuNTA0LjA4My0yLjIxOS4yMzcgNS4xMzktNS42NzYgMTIuMjUzLTkuMTkgMjAuMTE1LTkuMTkiLz48Y2lyY2xlIGN4PSIyNjkuOTMyIiBjeT0iMjczLjAzIiByPSI1MC42MTgiIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSI5LjIwMyIvPjxwYXRoIGZpbGw9IiNmZmYiIGQ9Ik0yNzYuODM0IDI0Mi4wOTJjMTUuNjcyIDAgMjguMzc2IDEzLjk2MyAyOC4zNzYgMzEuMTg4cy0xMi43MDQgMzEuMTg5LTI4LjM3NiAzMS4xODktMjguMzc3LTEzLjk2NC0yOC4zNzctMzEuMTg5YzAtMi42MjUuMjk3LTUuMTc0Ljg1My03LjYxIDEuNjA2IDMuNzI3IDUuMzEyIDYuMzM4IDkuNjI4IDYuMzM4IDUuNzg5IDAgMTAuNDgyLTQuNjkzIDEwLjQ4Mi0xMC40ODJzLTQuNjkzLTEwLjQ4MS0xMC40ODItMTAuNDgxYy0uNzYxIDAtMS41MDQuMDgzLTIuMjE5LjIzNyA1LjEzOS01LjY3NiAxMi4yNTQtOS4xOSAyMC4xMTUtOS4xOSIvPjxwYXRoIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSI5LjIwMyIgZD0iTTE3Ny4xMzUgMzQ1LjMyOGMuODk5LS4yNCAyLjgwMS0uMTk3IDYuMzA0LjQ2NCA2LjMxMSAxLjE5IDE2LjU4NSA0LjIyMSAyNi4xMjIgNC4yMjEgOS40NTIgMCAxNy4xMTgtMS45MjYgMjEuMzgzLTIuNjAzLjc3LS4xMjMgMS4zODUtLjE5OSAxLjg3OC0uMjM3LS4zNCAxMy44NDMtMTIuODkzIDI3LjcxLTI3Ljg2OCAyNy43MS0xNS4xNzIgMC0yNy44NzYtMTMuODM1LTI3Ljg3Ni0yOC44MTEgMC0uMzU3LjAyOS0uNTk2LjA1Ny0uNzQ0WiIvPjwvc3ZnPg==" alt="Performance Models">
+  </a>
+</p>
+<div align="center">
+
+<h1 style="color: #9334E9;">⚡ P-Q-Image-2.1</h1>
+
+<h2>Few-step LoRA adapters for Qwen-Image-2.1 — ComfyUI</h2>
+
+<h3>
+  <span style="color: #9334E9;">6 steps (v0.2), 5 or 8 steps (v0.1)</span>
+  &nbsp;·&nbsp;
+  <span style="color: #9334E9;">Up to 6.3x faster</span>
+  &nbsp;·&nbsp;
+  <span style="color: #9334E9;">No CFG</span>
+  &nbsp;·&nbsp;
+  Text-to-image and image editing
+</h3>
+
+</div>
+
+**P-Q-Image-2.1** is a set of LoRA adapters that let
+[`Qwen/Qwen-Image-2.1`](https://huggingface.co/Qwen/Qwen-Image-2.1) generate
+and edit images in only 6 steps (v0.2), or 5 or 8 steps (v0.1). The adapters load
+on top of the base model, so the pipeline, text encoder, and VAE stay unchanged.
+
+This repository is the ComfyUI export of the original repo
+[`PrunaAI/P-Q-Image-2.1`](https://huggingface.co/PrunaAI/P-Q-Image-2.1),
 plus example workflows for text-to-image and image editing.
 
 The adapters load with the native `LoraLoaderModelOnly` node on top of the
@@ -24,20 +72,66 @@ Qwen-Image-2.1 base checkpoint from
 [Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1).
 No custom nodes are needed. Update ComfyUI to a version with native Qwen-Image-2.1 support.
 
-## Files
+> **v0.2: two new 6-step adapters.**
+>
+> - An advanced distillation for more faithful generation and editing.
+> - The first distillation removing need for prompt upsampling into the LoRA, thus significantly reducing end-to-end generation.
+> The v0.1 adapters remain available.
+
+## 6-step variants
+
+Pick **6-steps** for faithful base-model behavior, or **6-steps-upsampled** for
+improved visual quality from a short prompt, without a separate prompt-upsampling call.
+
+| Variant | File | Steps | Trade-off |
+|---|---|:---:|---|
+| `6-steps` | `p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors` | 6 | **Advanced distillation.** Recommended for preserving inputs when editing. |
+| `6-steps-upsampled` | `p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors` | 6 | **Improved visual quality, shorter end-to-end generation.** The first distillation that removes the need for a separate prompt-upsampling step. |
+
+**6-steps** is an advanced 6-step distillation of the base model. It is intended
+as a drop-in replacement for the base model in existing generation and editing
+workflows, using the adapter's step count and sigma schedule. It improves on the
+v0.1 adapters in almost all categories and is particularly good at editing tasks
+where preserving the input matters.
+
+**6-steps-upsampled** distills the prompt upsampler recommended for
+Qwen-Image-2.1 into a 6-step LoRA. It improves visual quality without a separate
+prompt-upsampling call, which significantly reduces end-to-end generation time.
+It may not reproduce exactly what the actual prompt upsampler would do, and we
+will continue improving it.
+
+<details>
+<summary><b>See the 5-step vs 8-step adapters (v0.1)</b></summary>
+
+### v0.1
+
+Both adapters are v0.1. Pick one based on whether you need **quality** or **speed**.
+
+| Variant | File | Steps | Trade-off |
+|---|---|:---:|---|
+| `8-steps` | `p_q_image_2.1_8_steps_v0.1_comfyui_bf16.safetensors` | 8 | **Higher quality.** Recommended as the default. |
+| `5-steps` | `p_q_image_2.1_5_steps_v0.1_comfyui_bf16.safetensors` | 5 | **Higher speed**, with noticeably lower visual quality. |
+
+Each adapter is trained for its own sigma schedule. Load only one at a time.
+For the v0.2 adapters, do not reuse the v0.1 schedules.
+
+</details>
+
+## Prompts, editing, and resolution
 
-| File | Steps | Use |
-|---|:---:|---|
-| `p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors` | 6 | **Recommended.** Pure few-step distillation, best for editing. |
-| `p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors` | 6 | Distills prompt upsampling into the adapter for improved visual quality. |
-| `p_q_image_2.1_8_steps_v0.1_comfyui_bf16.safetensors` | 8 | v0.1, higher quality. |
-| `p_q_image_2.1_5_steps_v0.1_comfyui_bf16.safetensors` | 5 | v0.1, higher speed, lower visual quality. |
+For **6-steps**, use the same prompts and reference images you would use with the
+base model. For **6-steps-upsampled**, start with your original prompt; the adapter
+is intended to capture the benefit of prompt upsampling during generation.
+Detailed prompts still let you specify the subject, setting, lighting, and style.
 
-The weights are identical to the diffusers files in
-[PrunaAI/Pruna-Q-Image-2.1](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1).
-The ComfyUI copies store bf16 weights and explicit `.alpha` tensors, because
-ComfyUI does not read the PEFT `lora_alpha` from safetensors metadata. Without
-them, the v0.1 adapters would load at half strength.
+The adapters were trained at **1K resolution only**, with a mix of simple and
+upsampled prompts, text-to-image generation, and single- and multi-image editing
+with **up to 3 reference images**. Prompt upsampling is optional; detailed prompts
+usually give better results.
+
+Start at **1024 × 1024** and use at most **3 reference images** for editing.
+Higher resolutions (including 2K) and more reference images may work, but are
+outside the training coverage, and quality may vary.
 
 ## Setup
 
@@ -49,7 +143,7 @@ ComfyUI/
     ├── diffusion_models/
     │   └── qwen_image_2.1_int8_convrot.safetensors   (or qwen_image_2.1_bf16.safetensors)
     ├── loras/
-    │   └── p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors
+    │   └── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors
     ├── text_encoders/
     │   └── qwen3vl_8b_int8_convrot.safetensors
     └── vae/
@@ -66,21 +160,20 @@ Drag a file from [`workflows/`](workflows) into ComfyUI.
 
 | Workflow | Adapter |
 |---|---|
-| `pruna_q_image_2.1_v0.2_6step_t2i.json` | v0.2 basic (switch `lora_name` to upsampled) |
-| `pruna_q_image_2.1_v0.2_6step_image_edit.json` | v0.2 basic (switch `lora_name` to upsampled) |
-| `pruna_q_image_2.1_v0.1_8step_t2i.json` | v0.1 8-step |
-| `pruna_q_image_2.1_v0.1_8step_image_edit.json` | v0.1 8-step |
-| `pruna_q_image_2.1_v0.1_5step_t2i.json` | v0.1 5-step |
-| `pruna_q_image_2.1_v0.1_5step_image_edit.json` | v0.1 5-step |
+| `pruna_q_image_2.1_v0.2_6step_t2i.json` | v0.2 `6-steps` (switch `lora_name` to `6-steps-upsampled`) |
+| `pruna_q_image_2.1_v0.2_6step_image_edit.json` | v0.2 `6-steps` (switch `lora_name` to `6-steps-upsampled`) |
+| `pruna_q_image_2.1_v0.1_8step_t2i.json` | v0.1 `8-steps` |
+| `pruna_q_image_2.1_v0.1_8step_image_edit.json` | v0.1 `8-steps` |
+| `pruna_q_image_2.1_v0.1_5step_t2i.json` | v0.1 `5-steps` |
+| `pruna_q_image_2.1_v0.1_5step_image_edit.json` | v0.1 `5-steps` |
 
-They are built from the official Qwen-Image-2.1 Turbo templates. Inside each
-subgraph, `UNETLoader` feeds `LoraLoaderModelOnly` at strength 1.0, and
-`SamplerCustomAdvanced` runs euler with cfg 1 and a fixed `ManualSigmas` schedule.
+Each workflow loads one adapter at strength **1.0** and samples with `euler`,
+cfg **1**, and the matching `ManualSigmas` schedule.
 
 ## Sampling settings
 
-Each adapter is trained for its own sigma schedule. `ManualSigmas` passes values
-through unchanged, so the workflows contain the final values the sampler runs on.
+Use the sigma schedule that matches the adapter. Do not mix v0.1 and v0.2
+schedules.
 
 | Adapter | ManualSigmas |
 |---|---|
@@ -89,26 +182,52 @@ through unchanged, so the workflows contain the final values the sampler runs on
 | v0.1, 5 steps | `1.0, 0.94, 0.857143, 0.666667, 0.4, 0` |
 
 - **cfg 1**, no negative prompt, sampler `euler`, LoRA strength **1.0**.
-- **v0.1** runs its sigmas unshifted, matching the diffusers quickstart.
-- **v0.2** applies the base model's exponential time shift to the raw schedule
-  `1.0, 0.925, 0.85, 0.625, 0.375, 0.125`. The values above are shifted for
-  1024x1024 output (mu = 0.6935). The adapters were trained at 1K resolution only.
-
-## Rebuilding
-
-```bash
-python convert_to_comfyui.py --src ../Pruna-Q-Image-2.1 --dst .
-git clone --depth 1 https://github.com/Comfy-Org/workflow_templates
-python build_workflows.py --templates workflow_templates/templates --out workflows
-python verify_comfyui.py --comfyui /path/to/ComfyUI --src ../Pruna-Q-Image-2.1 --dst .
-```
-
-`verify_comfyui.py` runs ComfyUI's own LoRA key mapping against the Comfy-Org
-checkpoint layout and checks that every target is patched with the same delta as diffusers.
+- The adapters were trained at 1K resolution only.
 
 ## License
 
-The adapters follow the license of the base model. See [LICENSE](LICENSE) and
-[NOTICE](NOTICE), and review the
-[Qwen-Image-2.1 license](https://huggingface.co/Qwen/Qwen-Image-2.1) before using
-or redistributing the adapters.
+The end distilled model is based on Qwen-Image-2.1.
+
+P-Q-Image-2.1 is a derivative of Qwen-Image-2.1 and is distributed
+under the **Qwen RESEARCH LICENSE AGREEMENT**. Review its use restrictions
+before using or redistributing the adapters. See [LICENSE](LICENSE) and
+[NOTICE](NOTICE).
+
+Qwen is licensed under the Qwen RESEARCH LICENSE AGREEMENT, Copyright (c) 2026
+Hangzhou Tongyi Laboratory Technology Co., Ltd. All Rights Reserved.
+
+## What's next?
+
+- **Use P-Q-Image-2.1 to generate and edit images with Qwen-Image-2.1 in a few steps.**
+- Compress your own models with [Pruna](https://github.com/PrunaAI/pruna) and give us a ⭐️ for more efficiency!
+- Want to use our optimized image models right away? Check [P-Image-Ideogram](https://www.pruna.ai/p-image-ideogram) and [P-Image-Edit](https://www.pruna.ai/p-image-edit).
+
+<style>
+.model-button {
+  display: inline-flex;
+  flex-direction: row;
+  justify-content: center;
+  align-items: center;
+  gap: 8px;
+
+  padding: 8px 20px;
+  border: none;
+  border-radius: 8px;
+
+  background: #9334e9;
+  color: #ffffff;
+
+  font-size: 14px;
+  font-weight: 400;
+  line-height: 1;
+  text-decoration: none;
+  white-space: nowrap;
+  cursor: pointer;
+
+  box-sizing: border-box;
+  overflow: visible;
+  opacity: 1;
+}
+</style>
+
+<a href="https://dashboard.pruna.ai/login?utm_source=huggingface&utm_medium=model_card&utm_campaign=hf_traffic" class="model-button">Try our models</a>

build_workflows.py

@@ -1,206 +0,0 @@
-"""Build Pruna-Q-Image-2.1 ComfyUI workflows from the official Qwen-Image-2.1 Turbo templates.
-
-The Turbo templates already sample with ManualSigmas and cfg 1. Each workflow
-swaps the Turbo checkpoint for the base checkpoint, inserts a
-LoraLoaderModelOnly between the UNet loader and the Qwen-Image-2.1 cache, exposes
-`lora_name` on the subgraph node, and writes the adapter's sigma schedule.
-
-Usage:
-    git clone --depth 1 https://github.com/Comfy-Org/workflow_templates
-    python build_workflows.py --templates workflow_templates/templates --out workflows
-"""
-
-import argparse
-import copy
-import json
-import re
-import uuid
-from pathlib import Path
-
-REPO_URL = "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI"
-
-TEMPLATES = {
-    "t2i": ("image_qwen_image_2_1_turbo_t2i.json", "Text to Image"),
-    "image_edit": ("image_qwen_image_2_1_turbo_image_edit.json", "Image Edit"),
-}
-
-VARIANTS = {
-    "v0.2_6step": {
-        "lora": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
-        "alternatives": ["p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors"],
-        "steps": 6,
-        "sigmas": "1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0",
-    },
-    "v0.1_8step": {
-        "lora": "p_q_image_2.1_8_steps_v0.1_comfyui_bf16.safetensors",
-        "alternatives": [],
-        "steps": 8,
-        "sigmas": "1.0, 0.933333, 0.857143, 0.769231, 0.666667, 0.545455, 0.4, 0.222222, 0",
-    },
-    "v0.1_5step": {
-        "lora": "p_q_image_2.1_5_steps_v0.1_comfyui_bf16.safetensors",
-        "alternatives": [],
-        "steps": 5,
-        "sigmas": "1.0, 0.94, 0.857143, 0.666667, 0.4, 0",
-    },
-}
-
-TEXT_RENAMES = {
-    "qwen_image_2.1_turbo_bf16.safetensors": "qwen_image_2.1_bf16.safetensors",
-    "qwen_image_2.1_turbo_int8_convrot.safetensors": "qwen_image_2.1_int8_convrot.safetensors",
-    '\\"2.1 TURBO\\"': '\\"2.1 PRUNA\\"',
-}
-
-
-def node_of_type(nodes: list, node_type: str) -> dict:
-    matches = [n for n in nodes if n["type"] == node_type]
-    if len(matches) != 1:
-        raise ValueError(f"expected one {node_type}, found {len(matches)}")
-    return matches[0]
-
-
-def set_widget(node: dict, name: str, value, index: int) -> None:
-    node["widgets_values"][index] = value
-    if "widgets_values_named" in node:
-        node["widgets_values_named"][name] = value
-
-
-def lora_list(variant: dict) -> list:
-    return [variant["lora"], *variant["alternatives"]]
-
-
-def insert_lora(workflow: dict, subgraph: dict, proxy: dict, variant: dict) -> None:
-    state = subgraph["state"]
-    lora_id = state["lastNodeId"] + 1
-    lora_out_link = state["lastLinkId"] + 1
-    lora_name_link = state["lastLinkId"] + 2
-    state["lastNodeId"], state["lastLinkId"] = lora_id, lora_name_link
-    workflow["last_node_id"], workflow["last_link_id"] = lora_id, lora_name_link
-
-    unet = node_of_type(subgraph["nodes"], "UNETLoader")
-    cache = node_of_type(subgraph["nodes"], "QwenImage21Cache")
-    model_link_id = unet["outputs"][0]["links"][0]
-    model_link = next(l for l in subgraph["links"] if l["id"] == model_link_id)
-    model_link["target_id"], model_link["target_slot"] = lora_id, 0
-    cache["inputs"][0]["link"] = lora_out_link
-
-    lora_name_slot = len(subgraph["inputs"])
-    subgraph["inputs"].append({
-        "id": str(uuid.uuid4()),
-        "name": "lora_name",
-        "type": "COMBO",
-        "linkIds": [lora_name_link],
-        "pos": [subgraph["inputs"][-1]["pos"][0], subgraph["inputs"][-1]["pos"][1] + 20],
-    })
-    subgraph["links"] += [
-        {"id": lora_out_link, "origin_id": lora_id, "origin_slot": 0, "target_id": cache["id"], "target_slot": 0, "type": "MODEL"},
-        {"id": lora_name_link, "origin_id": -10, "origin_slot": lora_name_slot, "target_id": lora_id, "target_slot": 1, "type": "COMBO"},
-    ]
-
-    x, y = unet["pos"]
-    subgraph["nodes"].append({
-        "id": lora_id,
-        "type": "LoraLoaderModelOnly",
-        "title": "Load Pruna-Q-Image-2.1 LoRA",
-        "pos": [x, y + 120],
-        "size": [510, 82],
-        "flags": {},
-        "order": unet.get("order", 0) + 1,
-        "mode": 0,
-        "inputs": [
-            {"localized_name": "model", "name": "model", "type": "MODEL", "link": model_link_id},
-            {"localized_name": "lora_name", "name": "lora_name", "type": "COMBO", "widget": {"name": "lora_name"}, "link": lora_name_link},
-            {"localized_name": "strength_model", "name": "strength_model", "type": "FLOAT", "widget": {"name": "strength_model"}, "link": None},
-        ],
-        "outputs": [{"localized_name": "MODEL", "name": "MODEL", "type": "MODEL", "links": [lora_out_link]}],
-        "properties": {
-            "Node name for S&R": "LoraLoaderModelOnly",
-            "models": [{"name": name, "url": f"{REPO_URL}/resolve/main/{name}", "directory": "loras"} for name in lora_list(variant)],
-        },
-        "widgets_values": [variant["lora"], 1.0],
-        "widgets_values_named": {"lora_name": variant["lora"], "strength_model": 1.0},
-    })
-
-    proxy["inputs"].append({"label": "lora_name", "name": "lora_name", "type": "COMBO", "widget": {"name": "lora_name"}, "link": None})
-    proxy["widgets_values"].append(variant["lora"])
-    if "widgets_values_named" in proxy:
-        proxy["widgets_values_named"]["lora_name"] = variant["lora"]
-
-
-def update_notes(workflow: dict, variant: dict) -> None:
-    lora_links = "\n".join(f"- [{name}]({REPO_URL}/resolve/main/{name})" for name in lora_list(variant))
-    lora_tree = "".join(
-        f"│   │   {'└' if i == len(lora_list(variant)) - 1 else '├'}── {name}\n" for i, name in enumerate(lora_list(variant))
-    )
-    steps_note = (
-        f"- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for "
-        f"**{variant['steps']} steps**. The schedule is fixed in the ManualSigmas node (`{variant['sigmas']}`). "
-        f"Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter."
-    )
-    if variant["alternatives"]:
-        steps_note += (
-            "\n- lora_name: `basic` is the faithful 6-step distillation; `upsampled` distills prompt upsampling into the "
-            "adapter. Both use the same schedule."
-        )
-
-    for node in workflow["nodes"]:
-        if node["type"] != "MarkdownNote":
-            continue
-        text = node["widgets_values"][0]
-        text = re.sub(r"^- steps:.*$", lambda _: steps_note, text, flags=re.MULTILINE)
-        text = text.replace(
-            "**text_encoders**",
-            f"**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI]({REPO_URL}))\n\n{lora_links}\n\n**text_encoders**",
-        )
-        text = text.replace("│   ├── 📂 text_encoders/", f"│   ├── 📂 loras/\n{lora_tree}│   ├── 📂 text_encoders/")
-        set_widget(node, "text", text, 0)
-
-
-def build(template: dict, kind_label: str, variant_key: str, variant: dict) -> dict:
-    serialized = json.dumps(template)
-    for old, new in TEXT_RENAMES.items():
-        serialized = serialized.replace(old, new)
-    workflow = json.loads(serialized)
-    workflow["id"] = str(uuid.uuid4())
-    workflow["revision"] = 0
-
-    subgraph = workflow["definitions"]["subgraphs"][0]
-    subgraph["name"] = f"{kind_label} (Pruna-Q-Image-2.1 {variant_key.replace('_', ' ')})"
-    proxy = node_of_type(workflow["nodes"], subgraph["id"])
-
-    sigmas = node_of_type(subgraph["nodes"], "ManualSigmas")
-    sigmas["title"] = f"Manual Sigmas (Pruna-Q-Image-2.1 {variant['steps']}-step schedule)"
-    set_widget(sigmas, "sigmas", variant["sigmas"], 0)
-
-    guider = node_of_type(subgraph["nodes"], "CFGGuider")
-    set_widget(guider, "cfg", 1, 0)
-    if "cfg" in proxy.get("widgets_values_named", {}):
-        names = list(proxy["widgets_values_named"])
-        set_widget(proxy, "cfg", 1, names.index("cfg"))
-
-    save = node_of_type(workflow["nodes"], "SaveImageAdvanced")
-    save["widgets_values"][0] = "Pruna_Q_Image_2.1"
-
-    insert_lora(workflow, subgraph, proxy, variant)
-    update_notes(workflow, variant)
-    return workflow
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--templates", type=Path, required=True)
-    parser.add_argument("--out", type=Path, required=True)
-    args = parser.parse_args()
-    args.out.mkdir(parents=True, exist_ok=True)
-
-    for kind, (filename, kind_label) in TEMPLATES.items():
-        template = json.loads((args.templates / filename).read_text())
-        for variant_key, variant in VARIANTS.items():
-            workflow = build(copy.deepcopy(template), kind_label, variant_key, variant)
-            target = args.out / f"pruna_q_image_2.1_{variant_key}_{kind}.json"
-            target.write_text(json.dumps(workflow, indent=2, ensure_ascii=False) + "\n")
-            print(target)
-
-
-if __name__ == "__main__":
-    main()

convert_to_comfyui.py

@@ -1,89 +0,0 @@
-"""Convert the Pruna-Q-Image-2.1 PEFT LoRAs into ComfyUI-ready files.
-
-ComfyUI ignores safetensors metadata, so the PEFT `lora_alpha` of the v0.1
-adapters is written as explicit `.alpha` tensors. The v0.2 adapters already
-have alpha folded into B, so their `.alpha` equals the rank (scale 1.0).
-
-Usage:
-    python convert_to_comfyui.py --src ../Pruna-Q-Image-2.1 --dst .
-"""
-
-import argparse
-import json
-from pathlib import Path
-
-import torch
-from safetensors import safe_open
-from safetensors.torch import save_file
-
-ADAPTERS = {
-    "p_q_image_2.1_basic_6_steps_v0.2.safetensors": {"steps": 6, "version": "v0.2"},
-    "p_q_image_2.1_upsampled_6_steps_v0.2.safetensors": {"steps": 6, "version": "v0.2"},
-    "p_q_image_2.1_8_steps_v0.1.safetensors": {"steps": 8, "version": "v0.1"},
-    "p_q_image_2.1_5_steps_v0.1.safetensors": {"steps": 5, "version": "v0.1"},
-}
-
-# ComfyUI ManualSigmas values. v0.1 runs unshifted; v0.2 is the README schedule
-# after the base model's exponential time shift at 1024x1024 (mu = 0.6935).
-COMFYUI_SIGMAS = {
-    ("v0.1", 8): "1.0, 0.933333, 0.857143, 0.769231, 0.666667, 0.545455, 0.4, 0.222222, 0",
-    ("v0.1", 5): "1.0, 0.94, 0.857143, 0.666667, 0.4, 0",
-    ("v0.2", 6): "1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0",
-}
-
-PREFIX = "transformer."
-
-
-def lora_scale(metadata: dict) -> float:
-    """Return the alpha / rank factor ComfyUI must apply on top of B @ A."""
-    if "alpha folded" in metadata.get("scaling", ""):
-        return 1.0
-    config = json.loads(metadata["lora_adapter_metadata"])
-    return config["transformer.lora_alpha"] / config["transformer.r"]
-
-
-def convert(src: Path, dst: Path, info: dict) -> Path:
-    with safe_open(src, "pt") as f:
-        metadata = f.metadata() or {}
-        tensors = {k: f.get_tensor(k) for k in f.keys()}
-
-    scale = lora_scale(metadata)
-    out = {}
-    for key, tensor in tensors.items():
-        if not key.startswith(PREFIX):
-            raise ValueError(f"{src.name}: unexpected key {key}")
-        name = key[len(PREFIX):]
-        out[name] = tensor.to(torch.bfloat16).contiguous()
-        if name.endswith(".lora_A.weight"):
-            module = name[: -len(".lora_A.weight")]
-            rank = tensor.shape[0]
-            out[f"{module}.alpha"] = torch.tensor(scale * rank, dtype=torch.float32)
-
-    sigmas = COMFYUI_SIGMAS[(info["version"], info["steps"])]
-    out_metadata = {
-        "format": "pt",
-        "base_model": "Qwen/Qwen-Image-2.1",
-        "source_file": src.name,
-        "steps": str(info["steps"]),
-        "comfyui_manual_sigmas": sigmas,
-        "cfg": "1.0",
-        "lora_strength": "1.0",
-    }
-    target = dst / src.name.replace(".safetensors", "_comfyui_bf16.safetensors")
-    save_file(out, target, metadata=out_metadata)
-    return target
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--src", type=Path, required=True)
-    parser.add_argument("--dst", type=Path, required=True)
-    args = parser.parse_args()
-
-    for filename, info in ADAPTERS.items():
-        target = convert(args.src / filename, args.dst, info)
-        print(f"{filename} -> {target.name} ({target.stat().st_size / 1e6:.0f} MB)")
-
-
-if __name__ == "__main__":
-    main()

verify_comfyui.py

@@ -1,138 +0,0 @@
-"""Check that ComfyUI loads every LoRA target and applies the same delta as diffusers.
-
-Runs ComfyUI's own `model_lora_keys_unet` and `load_lora` against the layout of
-the Comfy-Org Qwen-Image-2.1 checkpoint. Only the checkpoint's safetensors
-header is downloaded; weights are stand-ins on the meta device.
-
-Usage:
-    python verify_comfyui.py --comfyui /path/to/ComfyUI --src ../Pruna-Q-Image-2.1 --dst .
-"""
-
-import argparse
-import json
-import logging
-import struct
-import sys
-import urllib.request
-from pathlib import Path
-from types import SimpleNamespace
-
-import torch
-from safetensors import safe_open
-from safetensors.torch import load_file
-
-CHECKPOINT_URL = "https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors"
-SAMPLE_MODULES = (
-    "transformer_blocks.0.attn.to_q",
-    "transformer_blocks.7.img_mlp.gate_layer",
-    "transformer_blocks.7.img_mlp.proj",
-    "transformer_blocks.31.img_mlp.out",
-    "img_in",
-    "proj_out",
-)
-
-
-def fetch_header(url: str) -> dict:
-    def read(start: int, end: int) -> bytes:
-        request = urllib.request.Request(url, headers={"Range": f"bytes={start}-{end}"})
-        return urllib.request.urlopen(request).read()
-
-    size = struct.unpack("<Q", read(0, 7))[0]
-    header = json.loads(read(8, 8 + size - 1))
-    header.pop("__metadata__", None)
-    return header
-
-
-def diffusers_scale(path: Path) -> float:
-    with safe_open(path, "pt") as f:
-        metadata = f.metadata() or {}
-    if "alpha folded" in metadata.get("scaling", ""):
-        return 1.0
-    config = json.loads(metadata["lora_adapter_metadata"])
-    return config["transformer.lora_alpha"] / config["transformer.r"]
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--comfyui", type=Path, required=True)
-    parser.add_argument("--src", type=Path, required=True)
-    parser.add_argument("--dst", type=Path, required=True)
-    args = parser.parse_args()
-
-    sys.path.insert(0, str(args.comfyui))
-    sys.argv = [sys.argv[0], "--cpu"]
-    import comfy.lora
-    import comfy.model_base
-
-    header = fetch_header(CHECKPOINT_URL)
-    state_dict = {
-        f"diffusion_model.{k}": torch.empty(v["shape"], device="meta")
-        for k, v in header.items()
-        if k.endswith(".weight")
-    }
-
-    class FakeQwenImage21(comfy.model_base.QwenImage21):
-        def __init__(self):
-            torch.nn.Module.__init__(self)
-            self.model_config = SimpleNamespace(unet_config={"image_model": "qwen_image21"})
-
-        def state_dict(self):
-            return state_dict
-
-    key_map = comfy.lora.model_lora_keys_unet(FakeQwenImage21(), {})
-
-    missing_logs = []
-    handler = logging.Handler()
-    handler.emit = lambda record: missing_logs.append(record.getMessage())
-    logging.getLogger().addHandler(handler)
-
-    failed = False
-    for converted in sorted(args.dst.glob("*_comfyui_bf16.safetensors")):
-        source = args.src / converted.name.replace("_comfyui_bf16", "")
-        lora = load_file(converted)
-        original = load_file(source)
-        scale = diffusers_scale(source)
-
-        missing_logs.clear()
-        patches = comfy.lora.load_lora(lora, key_map)
-        modules = {k[: -len(".lora_A.weight")] for k in lora if k.endswith(".lora_A.weight")}
-        not_loaded = [m for m in missing_logs if "not loaded" in m]
-
-        shape_errors = []
-        for module in modules:
-            target = key_map[module]
-            key, offset = (target, None) if isinstance(target, str) else target
-            delta_shape = (lora[f"{module}.lora_B.weight"].shape[0], lora[f"{module}.lora_A.weight"].shape[1])
-            weight_shape = tuple(state_dict[key].shape)
-            expected = (offset[2], weight_shape[1]) if offset else weight_shape
-            if delta_shape != expected:
-                shape_errors.append((module, delta_shape, expected))
-
-        max_err = 0.0
-        for module in SAMPLE_MODULES:
-            if f"{module}.lora_A.weight" not in lora:
-                continue
-            adapter = patches[key_map[module]]
-            up, down, alpha = adapter.weights[0].float(), adapter.weights[1].float(), adapter.weights[2]
-            comfy_delta = up @ down * (alpha / down.shape[0])
-            a = original[f"transformer.{module}.lora_A.weight"].float()
-            b = original[f"transformer.{module}.lora_B.weight"].float()
-            diffusers_delta = b @ a * scale
-            err = ((comfy_delta - diffusers_delta).abs().max() / diffusers_delta.abs().max()).item()
-            max_err = max(max_err, err)
-
-        ok = len(patches) == len(modules) and not not_loaded and not shape_errors and max_err < 1e-2
-        failed |= not ok
-        print(
-            f"{'OK  ' if ok else 'FAIL'} {converted.name}: {len(patches)}/{len(modules)} modules patched, "
-            f"{len(not_loaded)} unloaded keys, {len(shape_errors)} shape errors, "
-            f"max relative delta error {max_err:.2e} (diffusers scale {scale})"
-        )
-        for line in not_loaded[:5] + [str(e) for e in shape_errors[:5]]:
-            print("    ", line)
-
-    sys.exit(1 if failed else 0)
-
-
-if __name__ == "__main__":
-    main()

workflows/pruna_q_image_2.1_v0.2_6step_image_edit.json

@@ -192,7 +192,7 @@
         "auto",
         "default",
         "qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors",
-        "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors"
+        "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors"
       ],
       "widgets_values_named": {
         "resolution": 0,
@@ -213,7 +213,7 @@
         "device": "auto",
         "dtype": "default",
         "clip_name_1": "qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors",
-        "lora_name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors"
+        "lora_name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors"
       }
     },
     {
@@ -280,10 +280,10 @@
       "title": "Note: Usage",
       "properties": {},
       "widgets_values": [
-        "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\n- resolution is a total pixel budget (not width or height). Aspect ratio is preserved.\n- Official default is 1024. The model supports up to 2048.\n- This template starts at 0: no resize beyond a multiple of 32.\n- Output follows image_1. Extra refs can differ in size or aspect.\n- custom_size off: canvas comes from the encode latent (image_1).\n- custom_size on: use ResolutionSelector width/height. Keep it close to the resized image_1 size, or the edit can shift.\n\n## Reference images\n- Up to 10 reference images (image_1 to image_10).\n- Mention them in the prompt as `<image1>`, `<image2>`, ...\n- image_1 is the edit target. The rest are references.\n\n## Custom settings\nIf you need custom settings, enter or unpack the subgraph to edit it. See the [Subgraph](https://docs.comfy.org/interface/features/subgraph) guide.\n\n## Parameters\n- prompt: edit instruction. Use `<image1>` ... `<image10>`.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `basic` is the faithful 6-step distillation; `upsampled` distills prompt upsampling into the adapter. Both use the same schedule."
+        "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\n- resolution is a total pixel budget (not width or height). Aspect ratio is preserved.\n- Official default is 1024. The model supports up to 2048.\n- This template starts at 0: no resize beyond a multiple of 32.\n- Output follows image_1. Extra refs can differ in size or aspect.\n- custom_size off: canvas comes from the encode latent (image_1).\n- custom_size on: use ResolutionSelector width/height. Keep it close to the resized image_1 size, or the edit can shift.\n\n## Reference images\n- Up to 10 reference images (image_1 to image_10).\n- Mention them in the prompt as `<image1>`, `<image2>`, ...\n- image_1 is the edit target. The rest are references.\n\n## Custom settings\nIf you need custom settings, enter or unpack the subgraph to edit it. See the [Subgraph](https://docs.comfy.org/interface/features/subgraph) guide.\n\n## Parameters\n- prompt: edit instruction. Use `<image1>` ... `<image10>`.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `6-step` is the faithful distillation; `6-step-upsampled` distills prompt upsampling into the adapter. Both use the same schedule."
       ],
       "widgets_values_named": {
-        "text": "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\n- resolution is a total pixel budget (not width or height). Aspect ratio is preserved.\n- Official default is 1024. The model supports up to 2048.\n- This template starts at 0: no resize beyond a multiple of 32.\n- Output follows image_1. Extra refs can differ in size or aspect.\n- custom_size off: canvas comes from the encode latent (image_1).\n- custom_size on: use ResolutionSelector width/height. Keep it close to the resized image_1 size, or the edit can shift.\n\n## Reference images\n- Up to 10 reference images (image_1 to image_10).\n- Mention them in the prompt as `<image1>`, `<image2>`, ...\n- image_1 is the edit target. The rest are references.\n\n## Custom settings\nIf you need custom settings, enter or unpack the subgraph to edit it. See the [Subgraph](https://docs.comfy.org/interface/features/subgraph) guide.\n\n## Parameters\n- prompt: edit instruction. Use `<image1>` ... `<image10>`.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `basic` is the faithful 6-step distillation; `upsampled` distills prompt upsampling into the adapter. Both use the same schedule."
+        "text": "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\n- resolution is a total pixel budget (not width or height). Aspect ratio is preserved.\n- Official default is 1024. The model supports up to 2048.\n- This template starts at 0: no resize beyond a multiple of 32.\n- Output follows image_1. Extra refs can differ in size or aspect.\n- custom_size off: canvas comes from the encode latent (image_1).\n- custom_size on: use ResolutionSelector width/height. Keep it close to the resized image_1 size, or the edit can shift.\n\n## Reference images\n- Up to 10 reference images (image_1 to image_10).\n- Mention them in the prompt as `<image1>`, `<image2>`, ...\n- image_1 is the edit target. The rest are references.\n\n## Custom settings\nIf you need custom settings, enter or unpack the subgraph to edit it. See the [Subgraph](https://docs.comfy.org/interface/features/subgraph) guide.\n\n## Parameters\n- prompt: edit instruction. Use `<image1>` ... `<image10>`.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `6-step` is the faithful distillation; `6-step-upsampled` distills prompt upsampling into the adapter. Both use the same schedule."
       },
       "color": "#222",
       "bgcolor": "#000"
@@ -384,10 +384,10 @@
       "title": "Note: Model links",
       "properties": {},
       "widgets_values": [
-        "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Input Asset\n\n- [ballet_dancer_photography_minimalist_architecture.png](https://raw.githubusercontent.com/Comfy-Org/workflow_templates/refs/heads/main/input/ballet_dancer_photography_minimalist_architecture.png)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_bf16.safetensors) (16.33 GB)\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB) \n- [qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_bf16.safetensors\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors│   └── ��� vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
+        "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Input Asset\n\n- [ballet_dancer_photography_minimalist_architecture.png](https://raw.githubusercontent.com/Comfy-Org/workflow_templates/refs/heads/main/input/ballet_dancer_photography_minimalist_architecture.png)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_bf16.safetensors) (16.33 GB)\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB) \n- [qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_bf16.safetensors\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
       ],
       "widgets_values_named": {
-        "text": "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Input Asset\n\n- [ballet_dancer_photography_minimalist_architecture.png](https://raw.githubusercontent.com/Comfy-Org/workflow_templates/refs/heads/main/input/ballet_dancer_photography_minimalist_architecture.png)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_bf16.safetensors) (16.33 GB)\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB) \n- [qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_bf16.safetensors\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
+        "text": "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Input Asset\n\n- [ballet_dancer_photography_minimalist_architecture.png](https://raw.githubusercontent.com/Comfy-Org/workflow_templates/refs/heads/main/input/ballet_dancer_photography_minimalist_architecture.png)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_bf16.safetensors) (16.33 GB)\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB) \n- [qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_bf16.safetensors\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_i2i.int8_convrot.safetensors│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
       },
       "color": "#222",
       "bgcolor": "#000"
@@ -2205,23 +2205,23 @@
               "Node name for S&R": "LoraLoaderModelOnly",
               "models": [
                 {
-                  "name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
-                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
                   "directory": "loras"
                 },
                 {
-                  "name": "p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors",
-                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "name": "p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors",
+                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors",
                   "directory": "loras"
                 }
               ]
             },
             "widgets_values": [
-              "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+              "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
               1.0
             ],
             "widgets_values_named": {
-              "lora_name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+              "lora_name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
               "strength_model": 1.0
             }
           }

workflows/pruna_q_image_2.1_v0.2_6step_t2i.json

@@ -183,7 +183,7 @@
         "qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors",
         "auto",
         "default",
-        "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors"
+        "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors"
       ],
       "widgets_values_named": {
         "prompt": "Underground techno event poster: black background, a two-line headline in heavy white grotesque capitals where the first line reads \"QWEN IMAGE\" and the second line reads \"2.1 PRUNA\", both lines the same very large size and each spanning the full poster width edge to edge, stacked tightly one above the other in the upper half, the decimal point in \"2.1\" rendered as a clearly visible full-size square dot, a scattered pile of overlapping black-and-white halftone portrait photos showing a mixed cast of different people, an elderly man, a young woman, a child and a bearded man among them, arranged in two loose rows across the lower half, every face is razor sharp and crisply in focus with fine detailed halftone dots, high micro-contrast, thin white grid lines, small fluorescent orange accent shapes, two-colour risograph print texture, square 1:1 poster format, both headline lines equally prominent, spelling exactly correct letter by letter, no other readable words, no watermark",
@@ -201,7 +201,7 @@
         "clip_name_1": "qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors",
         "device": "auto",
         "dtype_1": "default",
-        "lora_name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors"
+        "lora_name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors"
       }
     },
     {
@@ -265,10 +265,10 @@
       "title": "Note: Model links",
       "properties": {},
       "widgets_values": [
-        "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB)\n- [qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors\n│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
+        "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB)\n- [qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors\n│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
       ],
       "widgets_values_named": {
-        "text": "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB)\n- [qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors\n│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
+        "text": "Guide: [Subgraph](https://docs.comfy.org/interface/features/subgraph)\n\n## Model Links\n\n- [Hugging Face:Comfy-Org/Qwen-Image-2.1](https://huggingface.co/Comfy-Org/Qwen-Image-2.1)\n- [ModelScope:Comfy-Org/Qwen-Image-2.1](https://modelscope.cn/models/Comfy-Org/Qwen-Image-2.1)\n\n**diffusion_models**\n\n- [qwen_image_2.1_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_bf16.safetensors) (13.25 GB)\n- [qwen_image_2.1_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/diffusion_models/qwen_image_2.1_int8_convrot.safetensors) (6.76 GB)\n\n**loras** ([PrunaAI/Pruna-Q-Image-2.1-ComfyUI](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI))\n\n- [p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors)\n- [p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors](https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors)\n\n**text_encoders**\n\n- [qwen3vl_8b_int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3vl_8b_int8_convrot.safetensors) (8.71 GB)\n- [qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/text_encoders/qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors) (8.82 GB)\n\n**vae**\n\n- [qwen_image_2.1_vae_bf16.safetensors](https://huggingface.co/Comfy-Org/Qwen-Image-2.1/resolve/main/vae/qwen_image_2.1_vae_bf16.safetensors) (644.2 MB)\n\n\n## Model Storage Location\n\n```\n📂 ComfyUI/\n├── 📂 models/\n│   ├── 📂 diffusion_models/\n│   │   ├── qwen_image_2.1_bf16.safetensors\n│   │   └── qwen_image_2.1_int8_convrot.safetensors\n│   ├── 📂 loras/\n│   │   ├── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors\n│   │   └── p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors\n│   ├── 📂 text_encoders/\n│   │   ├── qwen3vl_8b_int8_convrot.safetensors\n│   │   └── qwen3.5_9b_qwen_image_2.1_pe_t2i.int8_convrot.safetensors\n│   └── 📂 vae/\n│       └── qwen_image_2.1_vae_bf16.safetensors\n```\n\n## Report Issue\n\nNote: Please update ComfyUI first ([guide](https://docs.comfy.org/installation/update_comfyui)) and prepare required models. Desktop/Cloud updates follow stable releases, so some nightly-supported models may not be available yet.\n\n- Cannot run / runtime errors: [ComfyUI/issues](https://github.com/comfyanonymous/ComfyUI/issues)\n- UI / frontend issues: [ComfyUI_frontend/issues](https://github.com/Comfy-Org/ComfyUI_frontend/issues)\n- Workflow issues: [workflow_templates/issues](https://github.com/Comfy-Org/workflow_templates/issues)\n"
       },
       "color": "#222",
       "bgcolor": "#000"
@@ -292,10 +292,10 @@
       "title": "Note: Usage",
       "properties": {},
       "widgets_values": [
-        "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\nOutput size comes from ResolutionSelector (width/height). Default is 1 megapixel (1024*1024). For native 2K, set 1:1 and 4 megapixels (2048*2048). Prefer multiples of 32.\n\n## Transparent Image\n\nTo generate a transparent image, wrap the prompt like this and replace the middle with the subject:\n\n---\nThis is an RGBA format image with transparency. `[your description]`. The image has an alpha channel and a transparent background.\n\n---\nSave as PNG to keep the alpha channel.\n\n## Parameters\n\n- prompt: what to generate. For RGBA, see Note: transparent.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `basic` is the faithful 6-step distillation; `upsampled` distills prompt upsampling into the adapter. Both use the same schedule.\n"
+        "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\nOutput size comes from ResolutionSelector (width/height). Default is 1 megapixel (1024*1024). For native 2K, set 1:1 and 4 megapixels (2048*2048). Prefer multiples of 32.\n\n## Transparent Image\n\nTo generate a transparent image, wrap the prompt like this and replace the middle with the subject:\n\n---\nThis is an RGBA format image with transparency. `[your description]`. The image has an alpha channel and a transparent background.\n\n---\nSave as PNG to keep the alpha channel.\n\n## Parameters\n\n- prompt: what to generate. For RGBA, see Note: transparent.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `6-step` is the faithful distillation; `6-step-upsampled` distills prompt upsampling into the adapter. Both use the same schedule.\n"
       ],
       "widgets_values_named": {
-        "text": "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\nOutput size comes from ResolutionSelector (width/height). Default is 1 megapixel (1024*1024). For native 2K, set 1:1 and 4 megapixels (2048*2048). Prefer multiples of 32.\n\n## Transparent Image\n\nTo generate a transparent image, wrap the prompt like this and replace the middle with the subject:\n\n---\nThis is an RGBA format image with transparency. `[your description]`. The image has an alpha channel and a transparent background.\n\n---\nSave as PNG to keep the alpha channel.\n\n## Parameters\n\n- prompt: what to generate. For RGBA, see Note: transparent.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `basic` is the faithful 6-step distillation; `upsampled` distills prompt upsampling into the adapter. Both use the same schedule.\n"
+        "text": "## Size\nQwen Image 2.1 has native 2K (2048*2048) support for direct output.\n\nOutput size comes from ResolutionSelector (width/height). Default is 1 megapixel (1024*1024). For native 2K, set 1:1 and 4 megapixels (2048*2048). Prefer multiples of 32.\n\n## Transparent Image\n\nTo generate a transparent image, wrap the prompt like this and replace the middle with the subject:\n\n---\nThis is an RGBA format image with transparency. `[your description]`. The image has an alpha channel and a transparent background.\n\n---\nSave as PNG to keep the alpha channel.\n\n## Parameters\n\n- prompt: what to generate. For RGBA, see Note: transparent.\n- negative_prompt: unused while cfg is 1.\n- cfg: keep 1 for the Qwen Image 2.1 official path. Raise it only if you use a negative prompt.\n- steps: this workflow runs the Qwen Image 2.1 base checkpoint with a Pruna-Q-Image-2.1 LoRA distilled for **6 steps**. The schedule is fixed in the ManualSigmas node (`1.0, 0.961054, 0.918949, 0.769302, 0.545554, 0.222292, 0`). Keep LoRA strength 1.0, cfg 1 and euler, and change the schedule only together with a different adapter.\n- lora_name: `6-step` is the faithful distillation; `6-step-upsampled` distills prompt upsampling into the adapter. Both use the same schedule.\n"
       },
       "color": "#222",
       "bgcolor": "#000"
@@ -1648,23 +1648,23 @@
               "Node name for S&R": "LoraLoaderModelOnly",
               "models": [
                 {
-                  "name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
-                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
                   "directory": "loras"
                 },
                 {
-                  "name": "p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors",
-                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_upsampled_6_steps_v0.2_comfyui_bf16.safetensors",
+                  "name": "p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors",
+                  "url": "https://huggingface.co/PrunaAI/Pruna-Q-Image-2.1-ComfyUI/resolve/main/p_q_image_2.1_6_steps_upsampled_v0.2_comfyui_bf16.safetensors",
                   "directory": "loras"
                 }
               ]
             },
             "widgets_values": [
-              "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+              "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
               1.0
             ],
             "widgets_values_named": {
-              "lora_name": "p_q_image_2.1_basic_6_steps_v0.2_comfyui_bf16.safetensors",
+              "lora_name": "p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors",
               "strength_model": 1.0
             }
           }