ace.co/docs/peft  FlaxModelMixin  class diffusers.FlaxModelMixin  < source > ( )    Base class for all Flax models. FlaxModelMixin takes care of storing the model configuration and provides methods for loading, downloading and
saving models. config_name (str) — Filename to save a model to when calling save_pretrained().  from_pretrained  < source > ( pretrained_model_name_or_path: Union dtype: dtype = <class 'jax.numpy.float32'> *model_args **k