[core] pass diffusers library name to kernels calls (#14441)

up

Author: sayakpaul

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_flash_attn_3_available,
@@ -724,7 +725,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)
 

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
 

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. "