import spaces
import torch
from diffusers import DiffusionPipeline
from torchao import quantize_

pipeline = DiffusionPipeline.from_pretrained('Wan2.2')
quantize_(pipeline.transformer, Float8Config())
spaces.aoti_blocks_load(pipeline.transformer, 'zerogpu-aoti/Wan2', variant='fp8da')