—
Can be either:

A string, the repo id (for example runwayml/stable-diffusion-v1-5) of a pretrained pipeline
hosted on the Hub.
A path to a directory (for example ./my_model_directory) containing the model weights saved
using save_pretrained().
   dtype (str or jnp.dtype, optional) —
Override the default jnp.dtype and load the model under this dtype. If "auto", the dtype is
automatically derived from the model’s weights.   force_download (bool