From 1ccf86ce84e0ab1b83e8d5f24fb5a2117c233197 Mon Sep 17 00:00:00 2001 From: yuanwu Date: Thu, 22 May 2025 17:01:07 +0300 Subject: [PATCH] Use the 4.52.2 transformers Signed-off-by: yuanwu --- backends/gaudi/tgi-entrypoint.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/gaudi/tgi-entrypoint.sh b/backends/gaudi/tgi-entrypoint.sh index 3153870d..80ff1720 100644 --- a/backends/gaudi/tgi-entrypoint.sh +++ b/backends/gaudi/tgi-entrypoint.sh @@ -12,7 +12,7 @@ if [[ "$ATTENTION" == "paged" ]]; then # Check if Llama-4 is in the command line arguments if [[ "$*" == *"Llama-4"* || "$*" == *"Qwen3"* ]]; then echo 'ATTENTION=paged and Llama-4 or Qwen3 detected' - pip install transformers==4.52.1 + pip install transformers==4.52.2 fi fi