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

A string, the model id of a pretrained model hosted inside a model repo on huggingface.co.
Valid model ids should have an organization name, like google/ddpm-celebahq-256.
A path to a directory containing model weights saved using ~ModelMixin.save_config, e.g.,
./my_model_directory/.
A torch state
dict.



cache_dir (Union[str, os.PathLike], optional) —
Path to a directory in which a downloaded