How to use INCModel/Z-Image-tiny-for-testing with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("INCModel/Z-Image-tiny-for-testing", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
5f780f7 7ea4e3d 5f780f7
1
2
3
4
5
6
7
--- license: apache-2.0 --- This model is a tiny model of [Tongyi-MAI/Z-Image](https://huggingface.co/Tongyi-MAI/Z-Image) for testing. Please follow the license of the original model.