From 38c39b00e2c1220157176d9c88fda870bcdc2ce5 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Wed, 2 Oct 2024 16:07:33 +0200 Subject: [PATCH] Remove some stuff. --- .github/workflows/build.yaml | 3 +-- integration-tests/conftest.py | 4 ---- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 99844974..bfce447f 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -193,7 +193,6 @@ jobs: export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }} export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}" export HF_TOKEN=${{ secrets.HF_TOKEN }} - export ROCR_VISIBLE_DEVICES=0,1 + export ROCR_VISIBLE_DEVICES=3,4 echo $DOCKER_IMAGE - rocm-smi pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST} diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index b188e264..93ff9e40 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -534,10 +534,6 @@ def launcher(event_loop): yield ContainerLauncherHandle(client, container.name, port) - import time - - time.sleep(600) - if not use_flash_attention: del env["USE_FLASH_ATTENTION"]