File size: 324 Bytes
073b90c |
1 2 3 4 5 6 7 8 9 10 11 12 |
"""
Requirements:
- kernels package installed
- DIFFUSERS_ENABLE_HUB_KERNELS=true
"""
import torch
from diffusers import QwenImagePipeline
pipe = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.bfloat16).to(device)
spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/Qwen-Image", variant="fa3")
|