| """ | |
| 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") | |