add revision

Author: Abdennacer-Badaoui

backends/python/server/requirements-amd.txt

@@ -13,7 +13,8 @@ grpc-interceptor==0.15.3 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-reflection==1.58.0 ; python_version >= "3.9" and python_version < "3.13"
 grpcio-status==1.58.0 ; python_version >= "3.9" and python_version < "3.13"
 grpcio==1.58.0 ; python_version >= "3.9" and python_version < "3.13"
-huggingface-hub==0.24.5 ; python_version >= "3.9" and python_version < "3.13"
+huggingface-hub==0.34.4 ; python_version >= "3.9" and python_version < "3.13"
+hf-xet==1.5.1 ; python_version >= "3.9" and python_version < "3.13" and (platform_machine == "x86_64" or platform_machine == "amd64" or platform_machine == "arm64" or platform_machine == "aarch64")
 humanfriendly==10.0 ; python_version >= "3.9" and python_version < "3.13"
 idna==3.4 ; python_version >= "3.9" and python_version < "3.13"
 importlib-metadata==7.1.0 ; python_version >= "3.9" and python_version < "3.13"
@@ -58,3 +59,5 @@ yarl==1.9.4 ; python_version >= "3.9" and python_version < "3.13"
 zipp==3.18.1 ; python_version >= "3.9" and python_version < "3.13"
 pyrsistent==0.20.0 ; python_version >= "3.9" and python_version < "3.13"
 einops==0.8.0 ; python_version >= "3.9" and python_version < "3.13"
+kernels==0.12.3 ; python_version >= "3.9" and python_version < "3.13"
+tomli==2.4.1 ; python_version >= "3.9" and python_version < "3.11"

backends/python/server/text_embeddings_server/models/flash_bert.py

@@ -18,7 +18,9 @@
 if is_rocm():
     try:
         from kernels import get_kernel as _get_kernel
-        _triton_layer_norm = _get_kernel("kernels-community/triton-layer-norm")
+        _triton_layer_norm = _get_kernel(
+            "kernels-community/triton-layer-norm", revision="v0.1.0"
+        )
     except Exception:
         pass
 

backends/python/server/text_embeddings_server/models/flash_mistral.py

@@ -19,7 +19,9 @@
 if is_rocm():
     try:
         from kernels import get_kernel as _get_kernel
-        _triton_layer_norm = _get_kernel("kernels-community/triton-layer-norm")
+        _triton_layer_norm = _get_kernel(
+            "kernels-community/triton-layer-norm", revision="v0.1.0"
+        )
     except Exception:
         pass
 

backends/python/server/text_embeddings_server/models/flash_qwen3.py

@@ -21,7 +21,9 @@
 if is_rocm():
     try:
         from kernels import get_kernel as _get_kernel
-        _triton_layer_norm = _get_kernel("kernels-community/triton-layer-norm")
+        _triton_layer_norm = _get_kernel(
+            "kernels-community/triton-layer-norm", revision="v0.1.0"
+        )
     except Exception:
         pass
 

backends/python/server/text_embeddings_server/models/jinaBert_model.py

@@ -20,7 +20,9 @@
 if is_rocm():
     try:
         from kernels import get_kernel as _get_kernel
-        _triton_layer_norm = _get_kernel("kernels-community/triton-layer-norm")
+        _triton_layer_norm = _get_kernel(
+            "kernels-community/triton-layer-norm", revision="v0.1.0"
+        )
     except Exception:
         pass