<Tip warning={true}>  
Training with 8-bit and 4-bit weights are only supported for training *extra* parameters.  
</Tip>  
Check your memory footprint with the `get_memory_footprint` method:  
```py
print(model.get_memory_footprint())
```  
Quantized models can be loaded from the [`~ModelMixin.from_pretrained`] method without needing to specify the `quantization_config` parameters:  
```py
from diffusers import FluxTransformer2DModel, BitsAndBytesConfig