[TRTLLM] Remove gptneox from LCNC trtllm tests (#1725)

deepjavalibrary · Apr 3, 2024 · c82b131 · c82b131
1 parent cad0503
commit c82b131
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/lmi-no-code.yml b/.github/workflows/lmi-no-code.yml
@@ -208,6 +208,7 @@ jobs:
           python3 llm/client.py no_code mistral-7b
           docker rm -f $(docker ps -aq)
       - name: GPTNeox lmi container
+        if: ${{ matrix.container  == 'deepspeed' }}
         working-directory: tests/integration
         run: |
           rm -rf models