chore: remove pinned kernel versions

chunkable_ssd_combined.py

@@ -12,16 +12,16 @@ import triton.language as tl
 
 from einops import rearrange
 
-from transformers.integrations.hub_kernels import get_kernel
+from transformers.integrations.hub_kernels import lazy_load_kernel
 
 
-# Fixed revisions because kernels after 2026-04-14 do not expose the functions we need anymore.
-causal_conv1d = get_kernel("kernels-community/causal-conv1d", revision="dc7072f0e9d799b247a2517a909ebb209d50bea0")
-mamba_ssm = get_kernel("kernels-community/mamba-ssm", revision="00b2ecd499379f9bcf969b6796e53bc867f4ad38")
-
+causal_conv1d = lazy_load_kernel("causal-conv1d")
+assert causal_conv1d is not None, "Failed to load causal-conv1d kernel. Make sure all requirements are fulfilled: https://huggingface.co/kernels-community/causal-conv1d"
 causal_conv1d_fwd_function = causal_conv1d.cpp_functions.causal_conv1d_fwd_function
 causal_conv1d_bwd_function = causal_conv1d.cpp_functions.causal_conv1d_bwd_function
 
+mamba_ssm = lazy_load_kernel("mamba-ssm")
+assert mamba_ssm is not None, "Failed to load mamba-ssm kernel. Make sure all requirements are fulfilled: https://huggingface.co/kernels-community/mamba-ssm"
 custom_fwd = mamba_ssm.utils.torch.custom_fwd
 custom_bwd = mamba_ssm.utils.torch.custom_bwd
 _layer_norm_bwd = mamba_ssm.ops.triton.layernorm_gated._layer_norm_bwd