diff --git a/src/diffusers/models/attention_dispatch.py b/src/diffusers/models/attention_dispatch.py index 9414c151fd67..80ab1d89ba24 100644 --- a/src/diffusers/models/attention_dispatch.py +++ b/src/diffusers/models/attention_dispatch.py @@ -30,6 +30,7 @@ if torch.distributed.is_available(): import torch.distributed._functional_collectives as funcol +from .. import __version__ from ..utils import ( get_logger, is_aiter_available, @@ -728,7 +729,12 @@ def _maybe_download_kernel_for_backend(backend: AttentionBackendName) -> None: try: from kernels import get_kernel - kernel_module = get_kernel(config.repo_id, revision=config.revision, version=config.version) + kernel_module = get_kernel( + config.repo_id, + revision=config.revision, + version=config.version, + user_agent={"diffusers": __version__}, + ) if needs_kernel: config.kernel_fn = _resolve_kernel_attr(kernel_module, config.function_attr) diff --git a/src/diffusers/quantizers/gguf/utils.py b/src/diffusers/quantizers/gguf/utils.py index 2aa682ab002e..e9fd8db82f9a 100644 --- a/src/diffusers/quantizers/gguf/utils.py +++ b/src/diffusers/quantizers/gguf/utils.py @@ -20,6 +20,7 @@ import torch import torch.nn as nn +from ... import __version__ from ...utils import is_accelerate_available, is_kernels_available, is_kernels_version from ...utils.constants import DIFFUSERS_TRUST_REMOTE_KERNELS @@ -46,7 +47,7 @@ ) # `kernels<0.14.0` has no `trust_remote_code` argument and executes the downloaded code unconditionally. trust_kwargs = {"trust_remote_code": True} if is_kernels_version(">=", "0.14.0") else {} - ops = get_kernel("Isotr0py/ggml", **trust_kwargs) + ops = get_kernel("Isotr0py/ggml", user_agent={"diffusers": __version__}, **trust_kwargs) else: ops = None diff --git a/src/diffusers/quantizers/nunchaku/utils.py b/src/diffusers/quantizers/nunchaku/utils.py index cca40e2799eb..7f220dfc7e0f 100644 --- a/src/diffusers/quantizers/nunchaku/utils.py +++ b/src/diffusers/quantizers/nunchaku/utils.py @@ -8,6 +8,7 @@ import torch import torch.nn as nn +from ... import __version__ from ...utils import is_accelerate_available, is_kernels_available, is_kernels_version from ...utils.constants import DIFFUSERS_TRUST_REMOTE_KERNELS @@ -30,7 +31,9 @@ ) # `kernels<0.14.0` has no `trust_remote_code` argument and executes the downloaded code unconditionally. trust_kwargs = {"trust_remote_code": True} if is_kernels_version(">=", "0.14.0") else {} - ops = get_kernel(_HF_KERNEL_REPO, version=_HF_KERNEL_VERSION, **trust_kwargs).ops + ops = get_kernel( + _HF_KERNEL_REPO, version=_HF_KERNEL_VERSION, user_agent={"diffusers": __version__}, **trust_kwargs + ).ops else: raise ImportError( "Loading Nunchaku checkpoints requires the Hugging Face `kernels` package. "