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