File size: 285 Bytes
49d0222 |
1 2 3 4 5 6 7 8 9 |
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')
|