Z Image Turbo 安裝
若要安裝 Z Image Turbo,您需要使用最新版本的 diffusers。Z Image Turbo 安裝過程需要 Python 3.8+ 和 PyTorch。以下是如何啟動並執行 Z Image Turbo:
pip install git+https://github.com/huggingface/diffusersZ Image Turbo 程式碼範例
安裝 Z Image Turbo 後,您可以立即開始生成圖像。此 Z Image Turbo 程式碼範例示範了基本的使用模式。Z Image Turbo API 直觀且開發人員友善,使整合無縫。
import torch
from diffusers import ZImagePipeline
# Load Z Image Turbo
pipe = ZImagePipeline.from_pretrained(
"Tongyi-MAI/Z-Image-Turbo",
torch_dtype=torch.bfloat16,
)
pipe.to("cuda")
# Generate with Z Image Turbo
image = pipe(
prompt="Your prompt here",
height=1024,
width=1024,
num_inference_steps=9,
guidance_scale=0.0,
).images[0]Z Image Turbo 需求
在使用 Z Image Turbo 之前,請確保您的系統符合 Z Image Turbo 需求。Z Image Turbo 在啟用 CUDA 的 GPU 上表現最佳,但同時也支援 CPU 推論。為了獲得最佳的 Z Image Turbo 效能,我們建議使用 bfloat16 精度並在可用時啟用 Flash Attention。