diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index d65037ab..444ebf34 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -197,8 +197,8 @@ jobs:
           # export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
-          # python -c 'import docker; client = docker.from_env(); client.containers.run("registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm", devices=["/dev/kfd", "/dev/dri"], env={"MODEL_ID": "gpt2"}, detach=True)'
-          which pytest
+          python -c 'import docker; client = docker.from_env(); client.containers.run("registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm", devices=["/dev/kfd", "/dev/dri"], env={"MODEL_ID": "gpt2", "HF_TOKEN": ${HF_TOKEN}}, devices=["/dev/kfd", "/dev/dri"])'
+          # which pytest
           pip freeze
-          pytest -s -vv integration-tests -k flash_llama_grammar_tools_auto
+          # pytest -s -vv integration-tests -k flash_llama_grammar_tools_auto
           # pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST}