ll 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 **kwargs  )   Parameters   pretrained_model_name_or_path (str or os.PathLike) —
Can be either:

A strin