diff --git a/src/diffusers/modeling_flax_utils.py b/src/diffusers/modeling_flax_utils.py index 28cd29d226..071f88283d 100644 --- a/src/diffusers/modeling_flax_utils.py +++ b/src/diffusers/modeling_flax_utils.py @@ -27,7 +27,7 @@ from huggingface_hub import hf_hub_download from huggingface_hub.utils import EntryNotFoundError, RepositoryNotFoundError, RevisionNotFoundError from requests import HTTPError -from . import is_torch_available +from . import __version__, is_torch_available from .modeling_flax_pytorch_utils import convert_pytorch_state_dict_to_flax from .utils import ( CONFIG_NAME, @@ -289,7 +289,12 @@ class FlaxModelMixin: from_auto_class = kwargs.pop("_from_auto", False) subfolder = kwargs.pop("subfolder", None) - user_agent = {"file_type": "model", "framework": "flax", "from_auto_class": from_auto_class} + user_agent = { + "diffusers": __version__, + "file_type": "model", + "framework": "flax", + "from_auto_class": from_auto_class, + } # Load config if we don't provide a configuration config_path = config if config is not None else pretrained_model_name_or_path diff --git a/src/diffusers/modeling_utils.py b/src/diffusers/modeling_utils.py index 8bb5e728c1..5d28b8425a 100644 --- a/src/diffusers/modeling_utils.py +++ b/src/diffusers/modeling_utils.py @@ -26,6 +26,7 @@ from huggingface_hub import hf_hub_download from huggingface_hub.utils import EntryNotFoundError, RepositoryNotFoundError, RevisionNotFoundError from requests import HTTPError +from . import __version__ from .utils import CONFIG_NAME, DIFFUSERS_CACHE, HUGGINGFACE_CO_RESOLVE_ENDPOINT, WEIGHTS_NAME, logging @@ -297,7 +298,12 @@ class ModelMixin(torch.nn.Module): subfolder = kwargs.pop("subfolder", None) device_map = kwargs.pop("device_map", None) - user_agent = {"file_type": "model", "framework": "pytorch", "from_auto_class": from_auto_class} + user_agent = { + "diffusers": __version__, + "file_type": "model", + "framework": "pytorch", + "from_auto_class": from_auto_class, + } # Load config if we don't provide a configuration config_path = pretrained_model_name_or_path diff --git a/src/diffusers/pipeline_utils.py b/src/diffusers/pipeline_utils.py index 81118967aa..3ccdff9ba2 100644 --- a/src/diffusers/pipeline_utils.py +++ b/src/diffusers/pipeline_utils.py @@ -29,6 +29,7 @@ from huggingface_hub import snapshot_download from PIL import Image from tqdm.auto import tqdm +from . import __version__ from .configuration_utils import ConfigMixin from .dynamic_modules_utils import get_class_from_dynamic_module from .schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME @@ -372,6 +373,10 @@ class DiffusionPipeline(ConfigMixin): if custom_pipeline is not None: allow_patterns += [CUSTOM_PIPELINE_FILE_NAME] + user_agent = {"diffusers": __version__, "pipeline_class": config_dict["_class_name"]} + if custom_pipeline is not None: + user_agent["custom_pipeline"] = custom_pipeline + # download all allow_patterns cached_folder = snapshot_download( pretrained_model_name_or_path, @@ -382,6 +387,7 @@ class DiffusionPipeline(ConfigMixin): use_auth_token=use_auth_token, revision=revision, allow_patterns=allow_patterns, + user_agent=user_agent, ) else: cached_folder = pretrained_model_name_or_path