Skip to content

Commit 8c28d97

Browse files
authored
Lock transformers version (#93)
1 parent b41f716 commit 8c28d97

File tree

4 files changed

+8
-6
lines changed

4 files changed

+8
-6
lines changed

ci/L0_additional_outputs_vllm/test.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -30,6 +30,7 @@ source ../common/util.sh
3030

3131
pip3 install pytest==8.1.1
3232
pip3 install tritonclient[grpc]
33+
pip3 install transformers==4.53.3
3334

3435
# Prepare Model
3536
rm -rf models vllm_baseline_output.pkl && mkdir -p models

ci/L0_backend_vllm/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -28,7 +28,7 @@
2828
RET=0
2929
SUBTESTS="accuracy_test request_cancellation enabled_stream vllm_backend metrics_test"
3030

31-
python3 -m pip install tritonclient[grpc]
31+
python3 -m pip install tritonclient[grpc] transformers==4.53.3
3232

3333
for TEST in ${SUBTESTS}; do
3434
(cd ${TEST} && bash -ex test.sh && cd ..)

ci/L0_check_health_vllm/test.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -30,6 +30,7 @@ source ../common/util.sh
3030

3131
pip3 install pytest==8.1.1
3232
pip3 install tritonclient[grpc]
33+
pip3 install transformers==4.53.3
3334

3435
RET=0
3536

ci/L0_multi_gpu_vllm/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -28,7 +28,7 @@
2828
RET=0
2929
SUBTESTS="vllm_backend multi_lora"
3030

31-
python3 -m pip install tritonclient[grpc]
31+
python3 -m pip install tritonclient[grpc] transformers==4.53.3
3232

3333
for TEST in ${SUBTESTS}; do
3434
(cd ${TEST} && bash -ex test.sh && cd ..)

0 commit comments

Comments
 (0)