Method from_config

src/diffusers/models/auto_model.py:39–153

r""" Instantiate a model from a config dictionary or a pretrained model configuration file with random weights (no pretrained weights are loaded). Parameters: pretrained_model_name_or_path_or_dict (`str`, `os.PathLike`, or `dict`): Can be either:

(cls, pretrained_model_name_or_path_or_dict: str | os.PathLike | dict | None = None, **kwargs)

    @classmethod
    def from_config(cls, pretrained_model_name_or_path_or_dict: str | os.PathLike | dict | None = None, **kwargs):
        r"""
        Instantiate a model from a config dictionary or a pretrained model configuration file with random weights (no
        pretrained weights are loaded).

        Parameters:
            pretrained_model_name_or_path_or_dict (`str`, `os.PathLike`, or `dict`):
                Can be either:

                    - A string, the *model id* (for example `google/ddpm-celebahq-256`) of a pretrained model
                      configuration hosted on the Hub.
                    - A path to a *directory* (for example `./my_model_directory`) containing a model configuration
                      file.
                    - A config dictionary.

            cache_dir (`Union[str, os.PathLike]`, *optional*):
                Path to a directory where a downloaded pretrained model configuration is cached if the standard cache
                is not used.
            force_download (`bool`, *optional*, defaults to `False`):
                Whether or not to force the (re-)download of the model configuration, overriding the cached version if
                it exists.
            proxies (`Dict[str, str]`, *optional*):
                A dictionary of proxy servers to use by protocol or endpoint.
            local_files_only(`bool`, *optional*, defaults to `False`):
                Whether to only load local model configuration files or not.
            token (`str` or *bool*, *optional*):
                The token to use as HTTP bearer authorization for remote files.
            revision (`str`, *optional*, defaults to `"main"`):
                The specific model version to use.
            trust_remote_code (`bool`, *optional*, defaults to `False`):
                Whether to trust remote code.
            subfolder (`str`, *optional*, defaults to `""`):
                The subfolder location of a model file within a larger model repository on the Hub or locally.

        Returns:
            A model object instantiated from the config with random weights.

        Example:

        ```py
        from diffusers import AutoModel

        model = AutoModel.from_config("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="unet")
        ```
        """
        subfolder = kwargs.pop("subfolder", None)
        trust_remote_code = kwargs.pop("trust_remote_code", False)

        hub_kwargs_names = [
            "cache_dir",
            "force_download",
            "local_files_only",
            "proxies",
            "revision",
            "token",
        ]
        hub_kwargs = {name: kwargs.pop(name, None) for name in hub_kwargs_names}