diff --git a/.github/workflows/nv-v100-legacy.yml b/.github/workflows/nv-v100-legacy.yml index 1a35f1f1..cee04591 100644 --- a/.github/workflows/nv-v100-legacy.yml +++ b/.github/workflows/nv-v100-legacy.yml @@ -35,8 +35,8 @@ jobs: - name: Install dependencies run: | - pip install git+https://github.com/microsoft/DeepSpeed.git - pip install git+https://github.com/huggingface/transformers.git@v4.42.4 + pip install git+https://github.com/microsoft/DeepSpeed.git@lekurile/bloom_v_check + pip install git+https://github.com/huggingface/transformers.git pip install -U accelerate ds_report diff --git a/tests/legacy/test_local_deployment.py b/tests/legacy/test_local_deployment.py index 531036f6..69bac328 100644 --- a/tests/legacy/test_local_deployment.py +++ b/tests/legacy/test_local_deployment.py @@ -53,17 +53,6 @@ "query": "DeepSpeed is the greatest" }, ), - ( - "zero-shot-image-classification", - "openai/clip-vit-base-patch32", - { - "image": - "https://huggingface.co/datasets/Narsil/image_dummy/raw/main/parrots.png", - "candidate_labels": ["animals", - "humans", - "landscape"] - }, - ), ], ) def test_single_GPU(deployment, query): diff --git a/tests/legacy/test_non_persistent_deployment.py b/tests/legacy/test_non_persistent_deployment.py index ed2b13fb..fe5309d6 100644 --- a/tests/legacy/test_non_persistent_deployment.py +++ b/tests/legacy/test_non_persistent_deployment.py @@ -55,17 +55,6 @@ "query": "DeepSpeed is the greatest" }, ), - ( - "zero-shot-image-classification", - "openai/clip-vit-base-patch32", - { - "image": - "https://huggingface.co/datasets/Narsil/image_dummy/raw/main/parrots.png", - "candidate_labels": ["animals", - "humans", - "landscape"], - }, - ), ], ) def test_single_GPU(deployment, query):