Views
No views yet
1import torch
2from diffusers import ZImagePipeline # or ZImageImg2ImgPipeline / ZImageInpaintPipeline
3
4pipe = ZImagePipeline.from_pretrained(
5 "FINAL-Bench/POCKET-Image-Zimage", torch_dtype=torch.bfloat16
6).to("cuda")
7img = pipe("a serene mountain lake at sunrise, photorealistic", num_inference_steps=20).images[0]
8img.save("out.png")bitsandbytes (CUDA). Measured reload + generate peak: ~10.9 GB VRAM.
For Apple Silicon / CPU, an optimum-quanto int8 build (~13.4 GB) is the portable option.