From 0af4ac46b6321b6e66e257e296593ee2bb1b11c7 Mon Sep 17 00:00:00 2001
From: Yuwen Hu <yuwen.hu@intel.com>
Date: Mon, 20 Jan 2025 18:28:43 +0800
Subject: [PATCH] Small fix

---
 .github/workflows/llm_unit_tests.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 2c491977c59..7a93e684490 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -448,7 +448,7 @@ jobs:
 
           # TODO: Remove this workaround
           if [[ '${{ matrix.pytorch-version }}' == '2.6' ]]; then
-            python -m pip uninstall
+            python -m pip install transformers==4.37.0
           fi
 
       - name: Get Langchain version