From 8c28d97b9756623ebe701d4b92b97625a5223ea1 Mon Sep 17 00:00:00 2001
From: Misha Chornyi <99709299+mc-nv@users.noreply.github.com>
Date: Mon, 28 Jul 2025 14:08:12 -0700
Subject: [PATCH] Lock transformers version (#93)

---
 ci/L0_additional_outputs_vllm/test.sh | 3 ++-
 ci/L0_backend_vllm/test.sh            | 4 ++--
 ci/L0_check_health_vllm/test.sh       | 3 ++-
 ci/L0_multi_gpu_vllm/test.sh          | 4 ++--
 4 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/ci/L0_additional_outputs_vllm/test.sh b/ci/L0_additional_outputs_vllm/test.sh
index b5396701..75c679df 100755
--- a/ci/L0_additional_outputs_vllm/test.sh
+++ b/ci/L0_additional_outputs_vllm/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -30,6 +30,7 @@ source ../common/util.sh
 
 pip3 install pytest==8.1.1
 pip3 install tritonclient[grpc]
+pip3 install transformers==4.53.3
 
 # Prepare Model
 rm -rf models vllm_baseline_output.pkl && mkdir -p models
diff --git a/ci/L0_backend_vllm/test.sh b/ci/L0_backend_vllm/test.sh
index 074c0a8c..fb8f6eda 100755
--- a/ci/L0_backend_vllm/test.sh
+++ b/ci/L0_backend_vllm/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -28,7 +28,7 @@
 RET=0
 SUBTESTS="accuracy_test request_cancellation enabled_stream vllm_backend metrics_test"
 
-python3 -m pip install tritonclient[grpc]
+python3 -m pip install tritonclient[grpc] transformers==4.53.3
 
 for TEST in ${SUBTESTS}; do
     (cd ${TEST} && bash -ex test.sh && cd ..)
diff --git a/ci/L0_check_health_vllm/test.sh b/ci/L0_check_health_vllm/test.sh
index 50c1a097..8bb08010 100755
--- a/ci/L0_check_health_vllm/test.sh
+++ b/ci/L0_check_health_vllm/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -30,6 +30,7 @@ source ../common/util.sh
 
 pip3 install pytest==8.1.1
 pip3 install tritonclient[grpc]
+pip3 install transformers==4.53.3
 
 RET=0
 
diff --git a/ci/L0_multi_gpu_vllm/test.sh b/ci/L0_multi_gpu_vllm/test.sh
index 59123f13..072d3b56 100755
--- a/ci/L0_multi_gpu_vllm/test.sh
+++ b/ci/L0_multi_gpu_vllm/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -28,7 +28,7 @@
 RET=0
 SUBTESTS="vllm_backend multi_lora"
 
-python3 -m pip install tritonclient[grpc]
+python3 -m pip install tritonclient[grpc] transformers==4.53.3
 
 for TEST in ${SUBTESTS}; do
     (cd ${TEST} && bash -ex test.sh && cd ..)