| --- |
| license: apache-2.0 |
| tags: |
| - onnx |
| - webnn |
| - webgpu |
| - text-to-image |
| - z-image-turbo |
| --- |
| |
| # Z-Image-Turbo |
|
|
| This repository provides an optimized ONNX version of **[Tongyi-MAI/Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo)**, specifically tailored for efficient browser execution via **WebNN** and **WebGPU** using ONNX Runtime Web. |
|
|
| ## 🔄 Runtime Pipeline |
|
|
| The end-to-end inference flow runs sequentially across the ONNX sub-models under `onnx/` per image generation: |
|
|
| ### Inference Flow |
|
|
| <div align="center"> |
| <img src="https://cdn-uploads.huggingface.co/production/uploads/65ee995fc7b0ce53e4a91edb/2ckM0sTl3VVrRzJ-5recK.jpeg" style="max-width: 100%; height: auto;" alt="Z Image Turbo Pipeline"> |
| </div> |
|
|
| ### 🧩 Sub-model Components |
|
|
| | Stage | ONNX Model Artifact | Precision | Key Role | |
| |---|---|---|---| |
| | **1. Text Encoding** | `text_encoder_model_q4f16.onnx` | Q4F16 | Encodes text prompts into conditional embeddings | |
| | **2. Denoising** | `transformer_model_q4f16.onnx` | Q4F16 | Predicts noise latents across diffusion iterations | |
| | | `scheduler_step_model_f16.onnx` | FP16 | Computes denoised latents per step | |
| | **3. Decoding** | `vae_pre_process_model_f16.onnx` | FP16 | Scales and shifts latents for decoding | |
| | | `vae_decoder_model_f16.onnx` | FP16 | Reconstructs pixel-space RGB image from latents | |
| | **4. Safety** | `sc_prep_model_f16.onnx` | FP16 | Normalizes and pre-processes the image for the safety checker | |
| | | `safety_checker_model_f16.onnx` | FP16 | Inspects processed image features to ensure content safety | |
|
|
| ## 🛠️ Export & Reproduction |
|
|
| Instructions and scripts for exporting and optimizing the ONNX models from the original PyTorch weights can be found in the [model_exporter/README.md](model_exporter/README.md). |
|
|
| ## 🌐 Live Demo |
|
|
| Experience the model running in the browser: |
| * **Interactive Demo**: [WebNN Developer Preview - Z-Image-Turbo](https://microsoft.github.io/webnn-developer-preview/demos/z-image-turbo/) |
|
|
| ## 📑 Credits |
|
|
| * **Original PyTorch Model**: [Tongyi-MAI/Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) |
| * **Web Runtime**: [ONNX Runtime Web](https://github.com/microsoft/onnxruntime) / [WebNN Developer Preview](https://github.com/microsoft/webnn-developer-preview) |
|
|
