To serialize a quantized model in a given dtype, first load the model with the desired quantization dtype and then save it using the [`~ModelMixin.save_pretrained`] method.  
```python
import torch
from diffusers import FluxTransformer2DModel, TorchAoConfig