Upload 66 files

.gitattributes

@@ -33,3 +33,14 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/architecture.webp filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/decoupled-dmd.webp filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/DMDR.webp filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/leaderboard.png filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/reasoning.png filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/showcase_editing.png filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/showcase_realistic.png filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/showcase_rendering.png filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/showcase.jpg filter=lfs diff=lfs merge=lfs -text
+ZIT/assets/Z-Image-Gallery.pdf filter=lfs diff=lfs merge=lfs -text
+ZIT/tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text

ZIT/.cache/huggingface/.gitignore

@@ -0,0 +1 @@
+*

ZIT/.cache/huggingface/download/.gitattributes.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+c88e3c116b124ca91c94af1ac21d6d0db7deaa58
+1765723842.87288

ZIT/.cache/huggingface/download/README.md.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+c1f167cae8d334b21e9291b7d040bdd365a310a9
+1765723843.0835247

ZIT/.cache/huggingface/download/assets/DMDR.webp.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+2e6f3053b98d097f2aa11d3892bd9307326db41b65336bea54dc5825a0e03077
+1765723843.6776736

ZIT/.cache/huggingface/download/assets/Z-Image-Gallery.pdf.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+6f9895b3246d2547bac74bbe0be975da500eaae93f2cad4248ad3281786b1ac6
+1765723843.7332432

ZIT/.cache/huggingface/download/assets/architecture.webp.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+261af62ecc7e9749ae28e1d3a84e2f70a6c192d2017b7d8f020c7bff982ef59c
+1765723843.6776736

ZIT/.cache/huggingface/download/assets/decoupled-dmd.webp.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+4568ca559b997fc38f57dc1c3f5b1da3a3c144ae12419caa855ced972bf8c7aa
+1765723843.5603287

ZIT/.cache/huggingface/download/assets/leaderboard.png.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+e9fd4aa185bb7bff2b5515f2001b4d80df330595e78d6a098142e5a232bb4e4e
+1765723843.5428705

ZIT/.cache/huggingface/download/assets/leaderboard.webp.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+67cbfa6780584dce044144fb86afe495e3147abd
+1765723842.9801419

ZIT/.cache/huggingface/download/assets/reasoning.png.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+96c16b2c8d8dc67bb92ecc22d54b9955ab55136977f515bb76f4b2eb42eb3cdb
+1765723843.8256502

ZIT/.cache/huggingface/download/assets/showcase.jpg.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+f6ee74e066e00596e429f5a08140aebae1678e5935ce1e11ca6c1c6cd72432ee
+1765723843.7191947

ZIT/.cache/huggingface/download/assets/showcase_editing.png.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+7d720c3157fd0b0c1f07ac826c6d380b4bcb1b6933c64eb11bfe804ccf7c26f4
+1765723843.9885721

ZIT/.cache/huggingface/download/assets/showcase_realistic.png.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+697e6f6857f619314173508df72a14314cbb43e67475de7494123bb8b4f4eb2c
+1765723844.1912699

ZIT/.cache/huggingface/download/assets/showcase_rendering.png.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+3556dd66be2200d53f957424e12ecf914ddf3eded151cde86c7353f8b231284f
+1765723844.0754724

ZIT/.cache/huggingface/download/model_index.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+570c63a7c5580f63d0a8ed622324f50015dcf914
+1765723844.1922703

ZIT/.cache/huggingface/download/scheduler/scheduler_config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+2349bc43351afa730341d07dd44fce4ace1c4257
+1765723844.1681695

ZIT/.cache/huggingface/download/text_encoder/config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+e49eccdc32f36da9c09cfa0e737084f9e0105e5e
+1765723844.851518

ZIT/.cache/huggingface/download/text_encoder/generation_config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+20a8a9156fc8c3f25295ca067f61fdf120d517c5
+1765723844.1373098

ZIT/.cache/huggingface/download/text_encoder/model-00001-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+328a91d3122359d5547f9d79521205bc0a46e1f79a792dfe650e99fc2d651223
+1765723954.0234396

ZIT/.cache/huggingface/download/text_encoder/model-00002-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+6cd087b316306a68c562436b5492edbcf6e16c6dba3a1308279caa5a58e21ca5
+1765723942.6103716

ZIT/.cache/huggingface/download/text_encoder/model-00003-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+7ca841ee75b9c61267c0c6148fd8d096d3d21b6d3e161256a9b878154f91fc52
+1765723857.2813468

ZIT/.cache/huggingface/download/text_encoder/model.safetensors.index.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+95c0a0059df040d75dc6c396b174382cf61d2f91
+1765723844.8750381

ZIT/.cache/huggingface/download/tokenizer/merges.txt.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+31349551d90c7606f325fe0f11bbb8bd5fa0d7c7
+1765723845.914382

ZIT/.cache/huggingface/download/tokenizer/tokenizer.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+aeb13307a71acd8fe81861d94ad54ab689df773318809eed3cbe794b4492dae4
+1765723848.264321

ZIT/.cache/huggingface/download/tokenizer/tokenizer_config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+417d038a63fa3de29cfde265caedae14d1a58d92
+1765723844.6357412

ZIT/.cache/huggingface/download/tokenizer/vocab.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+4783fe10ac3adce15ac8f358ef5462739852c569
+1765723847.7064893

ZIT/.cache/huggingface/download/transformer/config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+c579190a5e03d602a2fd9647221c9d0d9441f150
+1765723845.354094

ZIT/.cache/huggingface/download/transformer/diffusion_pytorch_model-00001-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+95facd593e2549e8252acb571c653d57f7ddb7f1060d4e81712f152555a88804
+1765724011.9914029

ZIT/.cache/huggingface/download/transformer/diffusion_pytorch_model-00002-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+a4bbe43ee184a1fb5af4b412d27555f532893bdc3165b1149e304ed82b5d7015
+1765724012.0843074

ZIT/.cache/huggingface/download/transformer/diffusion_pytorch_model-00003-of-00003.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+aba4e37a590e63210878160a718d916d80398f4e1f78ab6c9b2b2a00d92769fa
+1765723986.1461906

ZIT/.cache/huggingface/download/transformer/diffusion_pytorch_model.safetensors.index.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+a3cadc1ed28e334ef58d4e6bbda107e25168ef47
+1765723848.283522

ZIT/.cache/huggingface/download/vae/config.json.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+894fa30ac1950cef422189d2d4cef11043c62875
+1765723848.7703586

ZIT/.cache/huggingface/download/vae/diffusion_pytorch_model.safetensors.metadata

@@ -0,0 +1,3 @@
+5f4b9cbb80cc95ba44fe6667dfd75710f7db2947
+f5b59a26851551b67ae1fe58d32e76486e1e812def4696a4bea97f16604d40a3
+1765723896.6316795

ZIT/.gitattributes

@@ -0,0 +1,46 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+assets/DMDR.webp filter=lfs diff=lfs merge=lfs -text
+assets/architecture.webp filter=lfs diff=lfs merge=lfs -text
+assets/decoupled-dmd.webp filter=lfs diff=lfs merge=lfs -text
+assets/reasoning.png filter=lfs diff=lfs merge=lfs -text
+assets/showcase.jpg filter=lfs diff=lfs merge=lfs -text
+assets/showcase_editing.png filter=lfs diff=lfs merge=lfs -text
+assets/showcase_realistic.png filter=lfs diff=lfs merge=lfs -text
+assets/showcase_rendering.png filter=lfs diff=lfs merge=lfs -text
+assets/Z-Image-Gallery.pdf filter=lfs diff=lfs merge=lfs -text
+assets/leaderboard.png filter=lfs diff=lfs merge=lfs -text

ZIT/README.md

@@ -0,0 +1,197 @@
+---
+license: apache-2.0
+language:
+- en
+pipeline_tag: text-to-image
+library_name: diffusers
+---
+
+
+<h1 align="center">⚡️- Image<br><sub><sup>An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer</sup></sub></h1>
+
+<div align="center">
+
+[![Official Site](https://img.shields.io/badge/Official%20Site-333399.svg?logo=homepage)](https://tongyi-mai.github.io/Z-Image-blog/)&#160;
+[![GitHub](https://img.shields.io/badge/GitHub-Z--Image-181717?logo=github&logoColor=white)](https://github.com/Tongyi-MAI/Z-Image)&#160;
+[![Hugging Face](https://img.shields.io/badge/%F0%9F%A4%97%20Checkpoint-Z--Image--Turbo-yellow)](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo)&#160;
+[![Hugging Face](https://img.shields.io/badge/%F0%9F%A4%97%20Online_Demo-Z--Image--Turbo-blue)](https://huggingface.co/spaces/Tongyi-MAI/Z-Image-Turbo)&#160;
+[![Hugging Face](https://img.shields.io/badge/%F0%9F%A4%97%20Mobile_Demo-Z--Image--Turbo-red)](https://huggingface.co/spaces/akhaliq/Z-Image-Turbo)&#160;
+[![ModelScope Model](https://img.shields.io/badge/🤖%20Checkpoint-Z--Image--Turbo-624aff)](https://www.modelscope.cn/models/Tongyi-MAI/Z-Image-Turbo)&#160;
+[![ModelScope Space](https://img.shields.io/badge/🤖%20Online_Demo-Z--Image--Turbo-17c7a7)](https://www.modelscope.cn/aigc/imageGeneration?tab=advanced&versionId=469191&modelType=Checkpoint&sdVersion=Z_IMAGE_TURBO&modelUrl=modelscope%253A%252F%252FTongyi-MAI%252FZ-Image-Turbo%253Frevision%253Dmaster%7D%7BOnline)&#160;
+[![Art Gallery PDF](https://img.shields.io/badge/%F0%9F%96%BC%20Art_Gallery-PDF-ff69b4)](assets/Z-Image-Gallery.pdf)&#160;
+[![Web Art Gallery](https://img.shields.io/badge/%F0%9F%8C%90%20Web_Art_Gallery-online-00bfff)](https://modelscope.cn/studios/Tongyi-MAI/Z-Image-Gallery/summary)&#160;
+<a href="https://arxiv.org/abs/2511.22699" target="_blank"><img src="https://img.shields.io/badge/Report-b5212f.svg?logo=arxiv" height="21px"></a>
+
+
+Welcome to the official repository for the Z-Image（造相）project!
+
+</div>
+
+
+
+## ✨ Z-Image
+
+Z-Image is a powerful and highly efficient image generation model with **6B** parameters. Currently there are three variants:
+
+- 🚀 **Z-Image-Turbo** – A distilled version of Z-Image that matches or exceeds leading competitors with only **8 NFEs** (Number of Function Evaluations). It offers **⚡️sub-second inference latency⚡️** on enterprise-grade H800 GPUs and fits comfortably within **16G VRAM consumer devices**. It excels in photorealistic image generation, bilingual text rendering (English & Chinese), and robust instruction adherence.
+
+- 🧱 **Z-Image-Base** – The non-distilled foundation model. By releasing this checkpoint, we aim to unlock the full potential for community-driven fine-tuning and custom development.
+
+- ✍️ **Z-Image-Edit** – A variant fine-tuned on Z-Image specifically for image editing tasks. It supports creative image-to-image generation with impressive instruction-following capabilities, allowing for precise edits based on natural language prompts.
+
+### 📥 Model Zoo
+
+| Model | Hugging Face                                                                                                                                                                                                                                                                                                              | ModelScope                                                                                                                                                                                                                                                                                                                                                                                                                                                                  |
+| :--- |:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| **Z-Image-Turbo** | [![Hugging Face](https://img.shields.io/badge/%F0%9F%A4%97%20Checkpoint%20-Z--Image--Turbo-yellow)](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) <br> [![Hugging Face Space](https://img.shields.io/badge/%F0%9F%A4%97%20Online%20Demo-Z--Image--Turbo-blue)](https://huggingface.co/spaces/Tongyi-MAI/Z-Image-Turbo) | [![ModelScope Model](https://img.shields.io/badge/🤖%20%20Checkpoint-Z--Image--Turbo-624aff)](https://www.modelscope.cn/models/Tongyi-MAI/Z-Image-Turbo) <br> [![ModelScope Space](https://img.shields.io/badge/%F0%9F%A4%96%20Online%20Demo-Z--Image--Turbo-17c7a7)](https://www.modelscope.cn/aigc/imageGeneration?tab=advanced&versionId=469191&modelType=Checkpoint&sdVersion=Z_IMAGE_TURBO&modelUrl=modelscope%3A%2F%2FTongyi-MAI%2FZ-Image-Turbo%3Frevision%3Dmaster) |
+| **Z-Image-Base** | *To be released*                                                                                                                                                                                                                                                                                                          | *To be released*                                                                                                                                                                                                                                                                                                                                                                                                                                                            |
+| **Z-Image-Edit** | *To be released*                                                                                                                                                                                                                                                                                                          | *To be released*                                                                                                                                                                                                                                                                                                                                                                                                                                                            |
+
+### 🖼️ Showcase
+
+📸 **Photorealistic Quality**: **Z-Image-Turbo** delivers strong photorealistic image generation while maintaining excellent aesthetic quality.
+
+![Showcase of Z-Image on Photo-realistic image Generation](assets/showcase_realistic.png)
+
+📖 **Accurate Bilingual Text Rendering**: **Z-Image-Turbo** excels at accurately rendering complex Chinese and English text.
+
+![Showcase of Z-Image on Bilingual Text Rendering](assets/showcase_rendering.png)
+
+💡  **Prompt Enhancing & Reasoning**: Prompt Enhancer empowers the model with reasoning capabilities, enabling it to transcend surface-level descriptions and tap into underlying world knowledge.
+
+![reasoning.jpg](assets/reasoning.png)
+
+🧠 **Creative Image Editing**: **Z-Image-Edit** shows a strong understanding of bilingual editing instructions, enabling imaginative and flexible image transformations.
+
+![Showcase of Z-Image-Edit on Image Editing](assets/showcase_editing.png)
+
+### 🏗️ Model Architecture
+We adopt a **Scalable Single-Stream DiT** (S3-DiT) architecture. In this setup, text, visual semantic tokens, and image VAE tokens are concatenated at the sequence level to serve as a unified input stream, maximizing parameter efficiency compared to dual-stream approaches.
+
+![Architecture of Z-Image and Z-Image-Edit](assets/architecture.webp)
+
+### 📈 Performance
+According to the Elo-based Human Preference Evaluation (on [*Alibaba AI Arena*](https://aiarena.alibaba-inc.com/corpora/arena/leaderboard?arenaType=T2I)), Z-Image-Turbo shows highly competitive performance against other leading models, while achieving state-of-the-art results among open-source models.
+
+<p align="center">
+  <a href="https://aiarena.alibaba-inc.com/corpora/arena/leaderboard?arenaType=T2I">
+    <img src="assets/leaderboard.png" alt="Z-Image Elo Rating on AI Arena"/><br />
+    <span style="font-size:1.05em; cursor:pointer; text-decoration:underline;"> Click to view the full leaderboard</span>
+  </a>
+</p>
+
+### 🚀 Quick Start
+Install the latest version of diffusers, use the following command:
+<details>
+  <summary><sup>Click here for details for why you need to install diffusers from source</sup></summary>
+
+  We have submitted two pull requests ([#12703](https://github.com/huggingface/diffusers/pull/12703) and [#12715](https://github.com/huggingface/diffusers/pull/12715)) to the 🤗 diffusers repository to add support for Z-Image. Both PRs have been merged into the latest official diffusers release.
+  Therefore, you need to install diffusers from source for the latest features and Z-Image support.
+
+</details>
+
+```bash
+pip install git+https://github.com/huggingface/diffusers
+```
+
+```python
+import torch
+from diffusers import ZImagePipeline
+
+# 1. Load the pipeline
+# Use bfloat16 for optimal performance on supported GPUs
+pipe = ZImagePipeline.from_pretrained(
+    "Tongyi-MAI/Z-Image-Turbo",
+    torch_dtype=torch.bfloat16,
+    low_cpu_mem_usage=False,
+)
+pipe.to("cuda")
+
+# [Optional] Attention Backend
+# Diffusers uses SDPA by default. Switch to Flash Attention for better efficiency if supported:
+# pipe.transformer.set_attention_backend("flash")    # Enable Flash-Attention-2
+# pipe.transformer.set_attention_backend("_flash_3") # Enable Flash-Attention-3
+
+# [Optional] Model Compilation
+# Compiling the DiT model accelerates inference, but the first run will take longer to compile.
+# pipe.transformer.compile()
+
+# [Optional] CPU Offloading
+# Enable CPU offloading for memory-constrained devices.
+# pipe.enable_model_cpu_offload()
+
+prompt = "Young Chinese woman in red Hanfu, intricate embroidery. Impeccable makeup, red floral forehead pattern. Elaborate high bun, golden phoenix headdress, red flowers, beads. Holds round folding fan with lady, trees, bird. Neon lightning-bolt lamp (⚡️), bright yellow glow, above extended left palm. Soft-lit outdoor night background, silhouetted tiered pagoda (西安大雁塔), blurred colorful distant lights."
+
+# 2. Generate Image
+image = pipe(
+    prompt=prompt,
+    height=1024,
+    width=1024,
+    num_inference_steps=9,  # This actually results in 8 DiT forwards
+    guidance_scale=0.0,     # Guidance should be 0 for the Turbo models
+    generator=torch.Generator("cuda").manual_seed(42),
+).images[0]
+
+image.save("example.png")
+```
+
+## 🔬 Decoupled-DMD: The Acceleration Magic Behind Z-Image
+
+[![arXiv](https://img.shields.io/badge/arXiv-2511.22677-b31b1b.svg)](https://arxiv.org/abs/2511.22677)
+
+Decoupled-DMD is the core few-step distillation algorithm that empowers the 8-step Z-Image model.
+
+Our core insight in Decoupled-DMD  is that the success of existing DMD (Distributaion Matching Distillation) methods is the result of two independent, collaborating mechanisms:
+
+-   **CFG Augmentation (CA)**: The primary **engine** 🚀 driving the distillation process, a factor largely overlooked in previous work.
+-   **Distribution Matching (DM)**: Acts more as a **regularizer** ⚖️, ensuring the stability and quality of the generated output.
+
+By recognizing and decoupling these two mechanisms, we were able to study and optimize them in isolation. This ultimately motivated us to develop an improved distillation process that significantly enhances the performance of few-step generation.
+
+![Diagram of Decoupled-DMD](assets/decoupled-dmd.webp)
+
+## 🤖 DMDR: Fusing DMD with Reinforcement Learning
+
+[![arXiv](https://img.shields.io/badge/arXiv-2511.13649-b31b1b.svg)](https://arxiv.org/abs/2511.13649)
+
+Building upon the strong foundation of Decoupled-DMD, our 8-step Z-Image model has already demonstrated exceptional capabilities. To achieve further improvements in terms of semantic alignment, aesthetic quality, and structural coherence—while producing images with richer high-frequency details—we present **DMDR**.
+
+Our core insight behind DMDR is that Reinforcement Learning (RL) and Distribution Matching Distillation (DMD) can be synergistically integrated during the post-training of few-step models. We demonstrate that:
+
+-   **RL Unlocks the Performance of DMD** 🚀
+-   **DMD Effectively Regularizes RL** ⚖️
+
+![Diagram of DMDR](assets/DMDR.webp)
+
+## ⏬ Download
+```bash
+pip install -U huggingface_hub
+HF_XET_HIGH_PERFORMANCE=1 hf download Tongyi-MAI/Z-Image-Turbo
+```
+
+## 📜 Citation
+
+If you find our work useful in your research, please consider citing:
+
+```bibtex
+@article{team2025zimage,
+  title={Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer},
+  author={Z-Image Team},
+  journal={arXiv preprint arXiv:2511.22699},
+  year={2025}
+}
+
+@article{liu2025decoupled,
+  title={Decoupled DMD: CFG Augmentation as the Spear, Distribution Matching as the Shield},
+  author={Dongyang Liu and Peng Gao and David Liu and Ruoyi Du and Zhen Li and Qilong Wu and Xin Jin and Sihan Cao and Shifeng Zhang and Hongsheng Li and Steven Hoi},
+  journal={arXiv preprint arXiv:2511.22677},
+  year={2025}
+}
+
+@article{jiang2025distribution,
+  title={Distribution Matching Distillation Meets Reinforcement Learning},
+  author={Jiang, Dengyang and Liu, Dongyang and Wang, Zanyi and Wu, Qilong and Jin, Xin and Liu, David and Li, Zhen and Wang, Mengmeng and Gao, Peng and Yang, Harry},
+  journal={arXiv preprint arXiv:2511.13649},
+  year={2025}
+}
+```

ZIT/assets/Z-Image-Gallery.pdf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f9895b3246d2547bac74bbe0be975da500eaae93f2cad4248ad3281786b1ac6
+size 15767436

ZIT/model_index.json

@@ -0,0 +1,24 @@
+{
+    "_class_name": "ZImagePipeline",
+    "_diffusers_version": "0.36.0.dev0",
+    "scheduler": [
+        "diffusers",
+        "FlowMatchEulerDiscreteScheduler"
+    ],
+    "text_encoder": [
+        "transformers",
+        "Qwen3Model"
+    ],
+    "tokenizer": [
+        "transformers",
+        "Qwen2Tokenizer"
+    ],
+    "transformer": [
+        "diffusers",
+        "ZImageTransformer2DModel"
+    ],
+    "vae": [
+        "diffusers",
+        "AutoencoderKL"
+    ]
+}

ZIT/scheduler/scheduler_config.json

@@ -0,0 +1,7 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.36.0.dev0",
+  "num_train_timesteps": 1000,
+  "use_dynamic_shifting": false,
+  "shift": 3.0
+}

ZIT/text_encoder/config.json

@@ -0,0 +1,30 @@
+{
+  "architectures": [
+    "Qwen3ForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "head_dim": 128,
+  "hidden_act": "silu",
+  "hidden_size": 2560,
+  "initializer_range": 0.02,
+  "intermediate_size": 9728,
+  "max_position_embeddings": 40960,
+  "max_window_layers": 36,
+  "model_type": "qwen3",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 36,
+  "num_key_value_heads": 8,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000,
+  "sliding_window": null,
+  "tie_word_embeddings": true,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.0",
+  "use_cache": true,
+  "use_sliding_window": false,
+  "vocab_size": 151936
+}