or_path_or_dict (str or os.PathLike or dict) —
Can be either:

A string, the model id (for example google/ddpm-celebahq-256) of a pretrained model hosted on
the Hub.
A path to a directory (for example ./my_model_directory) containing the model weights saved
with ModelMixin.save_pretrained().
A torch state
dict.
   cache_dir (Union[str, os.PathLike], optional) —
Path to a directory where a downloaded pretrained model configuration is cached if t