_name_or_path_or_dict: typing.Union[str, typing.Dict[str, torch.Tensor]]
**kwargs

)


Parameters 

pretrained_model_name_or_path_or_dict (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.,
.