ids are namespaced under a user or organization name, like
runwayml/stable-diffusion-v1-5.
A path to a directory containing model weights saved using save_pretrained(),
e.g., ./my_model_directory/.



dtype (jax.numpy.dtype, optional, defaults to jax.numpy.float32) —
The data type of the computation. Can be one of jax.numpy.float32, jax.numpy.float16 (on GPUs) and
jax.numpy.bfloat16 (on TPUs).
This can be used to enable mixed-precision training