Inspired from this example for FLUX.1-dev, with code taken from QwenImagePipeline in Diffusers.
import torch
from diffusers import AutoencoderKLQwenImage, QwenImagePipeline, QwenImageTransformer2DModel
from diffusers.image_processor import VaeImageProcessor
from torchao.quantization import float8_weight_only, quantize_
class QwenImagePromptEncoder: