Instructions to use PrunaAI/Pruna-Q-Image-2.1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use PrunaAI/Pruna-Q-Image-2.1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-2.1", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("PrunaAI/Pruna-Q-Image-2.1") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Update README.md
#1
by sdiazlor - opened
README.md
CHANGED
|
@@ -13,7 +13,6 @@ tags:
|
|
| 13 |
- distilled
|
| 14 |
- few-step
|
| 15 |
---
|
| 16 |
-
|
| 17 |
<!-- header start -->
|
| 18 |
<!-- 200823 -->
|
| 19 |
<div style="width: auto; margin-left: auto; margin-right: auto">
|
|
@@ -23,25 +22,25 @@ tags:
|
|
| 23 |
</div>
|
| 24 |
<!-- header end -->
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
<div align="center">
|
| 27 |
|
| 28 |
<h1 style="color: #9334E9;">⚡ Qwen-Image-2.1-P-Turbo</h1>
|
| 29 |
-
|
| 30 |
-
<
|
| 31 |
-
|
|
|
|
|
|
|
| 32 |
·
|
| 33 |
-
<span style="color: #9334E9;">
|
| 34 |
·
|
| 35 |
-
|
| 36 |
-
</
|
| 37 |
-
|
| 38 |
-
<p>
|
| 39 |
-
<a href="https://dashboard.pruna.ai/login?utm_source=huggingface&utm_medium=model_card&utm_campaign=hf_traffic"><img alt="Performance Models" src="https://img.shields.io/badge/Performance%20Models-Try%20them%20now-9334E9?style=for-the-badge&logo=data:image/svg%2bxml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHdpZHRoPSI0MTEiIGhlaWdodD0iNDc3IiBmaWxsPSJub25lIj48cGF0aCBmaWxsPSIjZmZmIiBkPSJNMjMuNzAzIDMwMC41MzJjLTEuODQyLS42NzQtNC4xNDItMS4wNTctNS43MDMtMi4wMjktMTIuNTU4LTcuODQ2LTguODM3LTI1LjE3MiA1LjI1NS0yOC4zNjFhLjY2LjY2IDAgMCAwIC4zNDgtLjIwMS42LjYgMCAwIDAgLjE1Ni0uMzZxNi40OS04MS45ODYgNjguNTc3LTEzNC42MjlhLjM0LjM0IDAgMCAwIC4wOTItLjEyNS4zMy4zMyAwIDAgMC0uMDI2LS4zLjMzLjMzIDAgMCAwLS4xMTItLjEwOGMtNC4xMTUtMi40OTctOC42NC00Ljc1LTEyLjM1My03LjQ0M2ExOTAgMTkwIDAgMCAxLTE4Ljc5NC0xNS41MjJjLTIuOTM2LTIuNzY4LTMuNzAzLTUuODkxLjU4LTcuNzdhLjUuNSAwIDAgMCAuMjgtLjYwOEM1MyA3NS45NSA0OC40MTcgNDguNTk5IDQ1LjUxOCAxOS42NXEtLjM5My0zLjg1Mi0uMjcxLTYuMzJhLjc1Ljc1IDAgMCAxIC4xMzQtLjQuNy43IDAgMCAxIC4zMjQtLjI1NXEyLjEwNC0uODQyIDUuODQ0LTEuNDMxYzU4LjExNC05LjEzNiAxMjcuMDA5IDguNDI1IDE0NC4zMDcgNzIuNjQ1cS4xMDMuMzY0LjE3OCAwIDUuMDk2LTI0LjA5NyA0LjYzOC00OC4xOTMtLjA0Ny0yLjc4NyAxLjg3OS00Ljc4OCAxMC40MDctMTAuOCAyNC43OTgtOC42ODdjNy40NzEgMS4wOTUgNy43NDIgNC42MzggNy4yMDkgMTEuMWEzMzQgMzM0IDAgMCAxLTExLjg4NCA2NC43MDYuMzg4LjM4OCAwIDAgMCAuMTgyLjQ0NC40LjQgMCAwIDAgLjE2NC4wNTFjOTIuNDIxIDguNDUzIDE1OC44NzUgNzkuNDg5IDE2NS4wODQgMTcxLjI3NGExIDEgMCAwIDAgLjEyOS40MjQuODMuODMgMCAwIDAgLjI5MS4yOTZjMy43NSAyLjE3IDYuMiAyLjIxNiA4Ljg1NSA2LjU5MiA1LjA2OCA4LjMzMiAxLjQ1IDIwLjQzMS04LjU5MyAyMi40N2ExLjcxIDEuNzEgMCAwIDAtMS4zNjUgMS40NjhxLTkuMTczIDc1LjM4My02NC43NjIgMTIzLjM3MWMtMjguNjIyIDI0LjcwNC02NS43NDQgMzkuNzQ5LTEwMy40NjQgNDIuMzMtMTAwLjM1OSA2Ljg2My0xODQuNDg2LTY1LjYyMi0xOTUuMDA1LTE2NS41N2EuOC44IDAgMCAwLS4xNTQtLjM5NS43NC43NCAwIDAgMC0uMzMzLS4yNU0xNTEuMzM4IDk3LjY0M2wxOS4wNjYgMTkuMzc1YS41LjUgMCAwIDAgLjE1LjEuNS41IDAgMCAwIC4xNzYuMDMxLjQzLjQzIDAgMCAwIC4zMTktLjE1cTEwLjg0Ny0xMi40NjUgNy40NDMtMjguMTU1Yy00LjUzNS0yMC44Ni0xOC41OTgtMzcuNjY0LTM3LjI3MS00OC4zODktMjEuMDY3LTEyLjY4OS00Ny44NjUtMTUuOTA1LTcyLjI4LTE0Ljg0cS0uNjE3LjAzLS4xNC40MjJsNjMuNDcyIDUzLjMyNmEuNC40IDAgMCAwIC4xNDQuMDcyLjM0LjM0IDAgMCAwIC4zMDEtLjA2OC4zLjMgMCAwIDAgLjA4OC0uMTI2IDQ3IDQ3IDAgMCAwIDMuMjgyLTEzLjA4MXEuMDIzLS4yMzMuMTI0LS40MzN0LjI2OC0uMzM0YzIuOTU1LTIuNDc4IDQuMTE1LjQ1OCA0LjY5NCAyLjg1MnEyLjMyIDkuNTM3LjE4NyAxOS4yNDRhMS40OCAxLjQ4IDAgMCAwIC40NCAxLjQwM3ptLTYuNTI2IDkuMjk1LTkuMDI0LTEwLjI3NmEuNy43IDAgMCAwLS4zNDMtLjIwNy44Ni44NiAwIDAgMC0uNDMzIDBxLTExLjY4IDMuMDIxLTIzLjExNC0uODIyYy01LjgxNi0zLjg1Mi0uNzU4LTUuNTE3IDIuNTktNS44MTZxNS45MTgtLjUzMyAxMS43MzUtMi41MDZhLjMuMyAwIDAgMCAuMTI3LS4wOC4zLjMgMCAwIDAgLjA3My0uMTMzLjM0LjM0IDAgMCAwIC4wMDEtLjE1NC4zNC4zNCAwIDAgMC0uMDctLjEzOGMtMTkuNDU5LTIxLjE1LTQwLjc2LTQwLjI0NS02Mi4yNzUtNTkuMjY0cS0uMzU2LS4zMTgtLjMxOC4xNWEzMDggMzA4IDAgMCAwIDguMzg3IDUxLjgwMmM0LjU5MiAxOC4zMDggMTMuNzU1IDM0LjE1NyAzMS44NDggNDEuODcycTI2LjI0NyAxMS4xODIgNTMuNzI5IDIuODMzYS41ODMuNTgzIDAgMCAwIC4yOTktLjg4OCAyMzUgMjM1IDAgMCAwLTEzLjIxMi0xNi4zNzNtNDUuMTI1IDQ5Ljc0NXEtMTIuNTk1IDEuMzE4LTIzLjQ0MS01LjU4M2MtMy4wNzctNC44OTkgMi4wNjYtOC41OTMgNS40NDItMTEuMTgzYS41ODQuNTg0IDAgMCAwIC4xODctLjcwMSA4LjQ2IDguNDYgMCAwIDAtMy40ODgtMy44NjIuODEuODEgMCAwIDAtLjgyMy4wMDljLTE2LjY5MSA5Ljk3Ny0zNC40ODUgMTAuNzgyLTUzLjE4NiA2Ljk2N3EtMS41Ny0uMzItMi44NTIuNjQ1LTYxLjQ1MSA0Ni4zNy02OS42NyAxMjMuOTEzLS42ODMgNi40MzMtMS44MDYgOC44MjdjLTIuOTQ1IDYuMzAyLTcuNzcgNy44NDUtMTQuMjIyIDkuNDcycS0uNDQ5LjExMyAwIC4yMzRjOS4yNzYgMi4zMSAxNC42OCA2LjE0MyAxNi4wNzQgMTYuNDAxIDguNDYyIDYyLjQ5IDQ4Ljk0IDExNy44NjQgMTEwLjM4MyAxMzguMzQxYTEwNCAxMDQgMCAwIDAgMTEuMDA2IDMuNTA3YzkyLjY0NSAyMy45MDkgMTgxLjIzMi0zMy42NjIgMjAyLjgwNC0xMjUuMDU0cTEuNjA5LTYuNzkgMy40NS0xOS4wODVjMS4xMDQtNy4zOTYgNS42OTUtMTIuMTE4IDEzLjE0Ny0xMy41NjguNDEyLS4wODQgMS45MDgtLjc1Ny4yODEtLjk5MS04LjA1MS0xLjE0MS0xMi4yNC02LjM4Ni0xMi45ODgtMTQuMTI4cS0xLjA3NS0xMS4yNS0yLjUxNi0xOS4wMzhjLTExLjc3Mi02My44OTItNTYuMzgzLTExNC41NzItMTE5LjI4NS0xMzEuNDZxLTE1LjEyLTMuODgtMzAuOTg3LTUuMjQ1YS4zODYuMzg2IDAgMCAwLS40MDIuMjUyIDE2OC42IDE2OC42IDAgMCAxLTExLjM1MiAyNC44NTRjLTQuMTMzIDcuNDA2LTguMTYzIDE1LjQ5NC0xNS43NTYgMTYuNDc2Ii8+PGNpcmNsZSBjeD0iMTQwLjA2NSIgY3k9IjI3My4wMyIgcj0iNTAuNjE4IiBzdHJva2U9IiNmZmYiIHN0cm9rZS13aWR0aD0iOS4yMDMiLz48cGF0aCBmaWxsPSIjZmZmIiBkPSJNMTQ2Ljk2NyAyNDIuMDkyYzE1LjY3MiAwIDI4LjM3NiAxMy45NjMgMjguMzc2IDMxLjE4OHMtMTIuNzA0IDMxLjE4OS0yOC4zNzYgMzEuMTg5LTI4LjM3Ny0xMy45NjQtMjguMzc3LTMxLjE4OWMwLTIuNjI1LjI5Ny01LjE3NC44NTItNy42MSAxLjYwNyAzLjcyNyA1LjMxMyA2LjMzOCA5LjYyOSA2LjMzOCA1Ljc4OSAwIDEwLjQ4MS00LjY5MyAxMC40ODItMTAuNDgycy00LjY5My0xMC40ODEtMTAuNDgyLTEwLjQ4MWMtLjc2MSAwLTEuNTA0LjA4My0yLjIxOS4yMzcgNS4xMzktNS42NzYgMTIuMjUzLTkuMTkgMjAuMTE1LTkuMTkiLz48Y2lyY2xlIGN4PSIyNjkuOTMyIiBjeT0iMjczLjAzIiByPSI1MC42MTgiIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSI5LjIwMyIvPjxwYXRoIGZpbGw9IiNmZmYiIGQ9Ik0yNzYuODM0IDI0Mi4wOTJjMTUuNjcyIDAgMjguMzc2IDEzLjk2MyAyOC4zNzYgMzEuMTg4cy0xMi43MDQgMzEuMTg5LTI4LjM3NiAzMS4xODktMjguMzc3LTEzLjk2NC0yOC4zNzctMzEuMTg5YzAtMi42MjUuMjk3LTUuMTc0Ljg1My03LjYxIDEuNjA2IDMuNzI3IDUuMzEyIDYuMzM4IDkuNjI4IDYuMzM4IDUuNzg5IDAgMTAuNDgyLTQuNjkzIDEwLjQ4Mi0xMC40ODJzLTQuNjkzLTEwLjQ4MS0xMC40ODItMTAuNDgxYy0uNzYxIDAtMS41MDQuMDgzLTIuMjE5LjIzNyA1LjEzOS01LjY3NiAxMi4yNTQtOS4xOSAyMC4xMTUtOS4xOSIvPjxwYXRoIHN0cm9rZT0iI2ZmZiIgc3Ryb2tlLXdpZHRoPSI5LjIwMyIgZD0iTTE3Ny4xMzUgMzQ1LjMyOGMuODk5LS4yNCAyLjgwMS0uMTk3IDYuMzA0LjQ2NCA2LjMxMSAxLjE5IDE2LjU4NSA0LjIyMSAyNi4xMjIgNC4yMjEgOS40NTIgMCAxNy4xMTgtMS45MjYgMjEuMzgzLTIuNjAzLjc3LS4xMjMgMS4zODUtLjE5OSAxLjg3OC0uMjM3LS4zNCAxMy44NDMtMTIuODkzIDI3LjcxLTI3Ljg2OCAyNy43MS0xNS4xNzIgMC0yNy44NzYtMTMuODM1LTI3Ljg3Ni0yOC44MTEgMC0uMzU3LjAyOS0uNTk2LjA1Ny0uNzQ0WiIvPjwvc3ZnPg==&logoColor=white"></a>
|
| 40 |
-
<a href="https://github.com/PrunaAI/pruna"><img alt="GitHub" src="https://img.shields.io/badge/GitHub-PrunaAI-9334E9?style=for-the-badge&logo=github&logoColor=white"></a>
|
| 41 |
-
<a href="https://discord.com/invite/JFQmtFKCjd"><img alt="Discord" src="https://img.shields.io/badge/Discord-Join%20us-9334E9?style=for-the-badge&logo=discord&logoColor=white"></a>
|
| 42 |
-
<a href="https://twitter.com/PrunaAI"><img alt="X" src="https://img.shields.io/badge/X-@PrunaAI-9334E9?style=for-the-badge&logo=x&logoColor=white"></a>
|
| 43 |
-
<a href="https://www.linkedin.com/company/pruna-ai"><img alt="LinkedIn" src="https://img.shields.io/badge/LinkedIn-PrunaAI-9334E9?style=for-the-badge&logo=linkedin&logoColor=white"></a>
|
| 44 |
-
</p>
|
| 45 |
|
| 46 |
</div>
|
| 47 |
|
|
@@ -55,6 +54,9 @@ on DMD. Improved using Qwen.**
|
|
| 55 |
> visual quality of the base model. We are still improving the distillation and
|
| 56 |
> will update this repository as new versions become available.
|
| 57 |
|
|
|
|
|
|
|
|
|
|
| 58 |
## Variants
|
| 59 |
|
| 60 |
Both adapters are v0.1. Pick one based on whether you need quality or speed.
|
|
@@ -73,7 +75,6 @@ Runs text-to-image and image editing with the same pipeline. Needs a **CUDA GPU*
|
|
| 73 |
```bash
|
| 74 |
pip install 'torch>=2.4.0' 'transformers>=5.17' accelerate peft pillow
|
| 75 |
pip install git+https://github.com/huggingface/diffusers@6256aa7666cedd47443adc8f82da9a10e110b09c
|
| 76 |
-
hf auth login # required while this repository is private
|
| 77 |
```
|
| 78 |
|
| 79 |
```python
|
|
@@ -111,8 +112,13 @@ settings = dict(
|
|
| 111 |
true_cfg_scale=1.0,
|
| 112 |
use_kv_cache=False,
|
| 113 |
)
|
|
|
|
|
|
|
|
|
|
| 114 |
|
| 115 |
-
|
|
|
|
|
|
|
| 116 |
image = pipe(
|
| 117 |
prompt=(
|
| 118 |
'A glowing neon shop sign that reads "QWEN IMAGE 2.1", mounted on a brick wall '
|
|
@@ -125,8 +131,11 @@ image = pipe(
|
|
| 125 |
**settings,
|
| 126 |
).images[0]
|
| 127 |
image.save("t2i_example.png")
|
|
|
|
|
|
|
|
|
|
| 128 |
|
| 129 |
-
|
| 130 |
image = pipe(
|
| 131 |
prompt="Change the background to a sunset beach",
|
| 132 |
image=Image.open("input.png").convert("RGB"),
|
|
@@ -155,9 +164,6 @@ image.save("edit_example.png")
|
|
| 155 |
- The 5-step adapter is faster than the 8-step one, but its images are
|
| 156 |
visibly worse.
|
| 157 |
- Short or vague text-to-image prompts give weaker results.
|
| 158 |
-
- Trained only at 1K resolution (around 1024 × 1024 pixels). Other resolutions
|
| 159 |
-
may give worse results. The next version will also be trained at higher
|
| 160 |
-
resolutions.
|
| 161 |
- The adapters are only intended for the settings above. Other step counts,
|
| 162 |
schedules or CFG values are not supported.
|
| 163 |
|
|
@@ -180,5 +186,34 @@ Hangzhou Tongyi Laboratory Technology Co., Ltd. All Rights Reserved.
|
|
| 180 |
|
| 181 |
- **Use P-Turbo to generate and edit images with Qwen-Image-2.1 in a few steps.**
|
| 182 |
- Compress your own models with [Pruna](https://github.com/PrunaAI/pruna) and give us a ⭐️ for more efficiency!
|
| 183 |
-
- Want to use our optimized models right away?
|
| 184 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
- distilled
|
| 14 |
- few-step
|
| 15 |
---
|
|
|
|
| 16 |
<!-- header start -->
|
| 17 |
<!-- 200823 -->
|
| 18 |
<div style="width: auto; margin-left: auto; margin-right: auto">
|
|
|
|
| 22 |
</div>
|
| 23 |
<!-- header end -->
|
| 24 |
|
| 25 |
+
[](https://github.com/PrunaAI/pruna)
|
| 26 |
+
[](https://twitter.com/PrunaAI)
|
| 27 |
+
[](https://www.linkedin.com/company/pruna-ai)
|
| 28 |
+
[](https://discord.com/invite/JFQmtFKCjd)
|
| 29 |
+
[](https://dashboard.pruna.ai/login?utm_source=huggingface&utm_medium=org_card&utm_campaign=hf_traffic)
|
| 30 |
+
|
| 31 |
<div align="center">
|
| 32 |
|
| 33 |
<h1 style="color: #9334E9;">⚡ Qwen-Image-2.1-P-Turbo</h1>
|
| 34 |
+
|
| 35 |
+
<h2>Few-step LoRA adapters for Qwen-Image-2.1</h2>
|
| 36 |
+
|
| 37 |
+
<h3>
|
| 38 |
+
<span style="color: #9334E9;">5 or 8 steps</span>
|
| 39 |
·
|
| 40 |
+
<span style="color: #9334E9;">No CFG</span>
|
| 41 |
·
|
| 42 |
+
Text-to-image and image editing
|
| 43 |
+
</h3>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
|
| 45 |
</div>
|
| 46 |
|
|
|
|
| 54 |
> visual quality of the base model. We are still improving the distillation and
|
| 55 |
> will update this repository as new versions become available.
|
| 56 |
|
| 57 |
+
## Examples
|
| 58 |
+
|
| 59 |
+
|
| 60 |
## Variants
|
| 61 |
|
| 62 |
Both adapters are v0.1. Pick one based on whether you need quality or speed.
|
|
|
|
| 75 |
```bash
|
| 76 |
pip install 'torch>=2.4.0' 'transformers>=5.17' accelerate peft pillow
|
| 77 |
pip install git+https://github.com/huggingface/diffusers@6256aa7666cedd47443adc8f82da9a10e110b09c
|
|
|
|
| 78 |
```
|
| 79 |
|
| 80 |
```python
|
|
|
|
| 112 |
true_cfg_scale=1.0,
|
| 113 |
use_kv_cache=False,
|
| 114 |
)
|
| 115 |
+
```
|
| 116 |
+
|
| 117 |
+
## For Text-to-Image
|
| 118 |
|
| 119 |
+
Longer, more descriptive prompts give better results.
|
| 120 |
+
|
| 121 |
+
```python
|
| 122 |
image = pipe(
|
| 123 |
prompt=(
|
| 124 |
'A glowing neon shop sign that reads "QWEN IMAGE 2.1", mounted on a brick wall '
|
|
|
|
| 131 |
**settings,
|
| 132 |
).images[0]
|
| 133 |
image.save("t2i_example.png")
|
| 134 |
+
```
|
| 135 |
+
|
| 136 |
+
### For Image Editing
|
| 137 |
|
| 138 |
+
```python
|
| 139 |
image = pipe(
|
| 140 |
prompt="Change the background to a sunset beach",
|
| 141 |
image=Image.open("input.png").convert("RGB"),
|
|
|
|
| 164 |
- The 5-step adapter is faster than the 8-step one, but its images are
|
| 165 |
visibly worse.
|
| 166 |
- Short or vague text-to-image prompts give weaker results.
|
|
|
|
|
|
|
|
|
|
| 167 |
- The adapters are only intended for the settings above. Other step counts,
|
| 168 |
schedules or CFG values are not supported.
|
| 169 |
|
|
|
|
| 186 |
|
| 187 |
- **Use P-Turbo to generate and edit images with Qwen-Image-2.1 in a few steps.**
|
| 188 |
- Compress your own models with [Pruna](https://github.com/PrunaAI/pruna) and give us a ⭐️ for more efficiency!
|
| 189 |
+
- 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).
|
| 190 |
+
|
| 191 |
+
<style>
|
| 192 |
+
.model-button {
|
| 193 |
+
display: inline-flex;
|
| 194 |
+
flex-direction: row;
|
| 195 |
+
justify-content: center;
|
| 196 |
+
align-items: center;
|
| 197 |
+
gap: 8px;
|
| 198 |
+
|
| 199 |
+
padding: 8px 20px;
|
| 200 |
+
border: none;
|
| 201 |
+
border-radius: 8px;
|
| 202 |
+
|
| 203 |
+
background: #9334e9;
|
| 204 |
+
color: #ffffff;
|
| 205 |
+
|
| 206 |
+
font-size: 14px;
|
| 207 |
+
font-weight: 400;
|
| 208 |
+
line-height: 1;
|
| 209 |
+
text-decoration: none;
|
| 210 |
+
white-space: nowrap;
|
| 211 |
+
cursor: pointer;
|
| 212 |
+
|
| 213 |
+
box-sizing: border-box;
|
| 214 |
+
overflow: visible;
|
| 215 |
+
opacity: 1;
|
| 216 |
+
}
|
| 217 |
+
</style>
|
| 218 |
+
|
| 219 |
+
<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>
|