Small fix

Author: Oscilloscope98

.github/workflows/llm_unit_tests.yml

@@ -425,15 +425,11 @@ jobs:
           bash python/llm/test/run-llm-inference-tests-gpu.sh
 
       - name: Run LLM example tests
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: remove this workaround
         shell: bash
         run: |
           python -m pip uninstall datasets -y
-          # TODO: Remove this workaround
-          if [[ '${{ matrix.pytorch-version }}' == '2.6' ]]; then
-            python -m pip install transformers==4.42.0 datasets peft==0.10.0
-          else
-            python -m pip install transformers==4.36.0 datasets peft==0.10.0
-          fi
+          python -m pip install transformers==4.36.0 datasets peft==0.10.0
 
           python -m pip install bitsandbytes scipy
           # Specific oneapi position on arc ut test machines
@@ -446,11 +442,6 @@ jobs:
           fi
           bash python/llm/test/run-llm-example-tests-gpu.sh
 
-          # TODO: Remove this workaround
-          if [[ '${{ matrix.pytorch-version }}' == '2.6' ]]; then
-            python -m pip install transformers==4.37.0
-          fi
-
       - name: Get Langchain version
         shell: bash
         id: get_langchain_version