From 6e305dcd7c79f466b78c7fe759ea959244368d35 Mon Sep 17 00:00:00 2001 From: dan Date: Tue, 26 Nov 2024 13:03:03 -0600 Subject: [PATCH] update yml and remove the weight fixture until required Signed-off-by: dan --- .github/workflows/pkgci_regression_test.yml | 4 ++-- .../shark-test-suite-models/llama/test_llama.py | 5 ----- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/.github/workflows/pkgci_regression_test.yml b/.github/workflows/pkgci_regression_test.yml index f21c0576ddbd..736a90fd849f 100644 --- a/.github/workflows/pkgci_regression_test.yml +++ b/.github/workflows/pkgci_regression_test.yml @@ -208,11 +208,11 @@ jobs: env: ROCM_CHIP: ${{ matrix.rocm-chip }} - - name: "Running SD3 special model tests" + - name: "Running llama special model tests" if: "!cancelled()" run: | source ${VENV_DIR}/bin/activate - pytest ./experimental/regression_suite/shark-test-suite-models/sd3 \ + pytest ./experimental/regression_suite/shark-test-suite-models/llama \ -k ${{ matrix.backend }} \ -rpfE \ --capture=no \ diff --git a/experimental/regression_suite/shark-test-suite-models/llama/test_llama.py b/experimental/regression_suite/shark-test-suite-models/llama/test_llama.py index 04b6b176de9b..37c2e0a042c6 100644 --- a/experimental/regression_suite/shark-test-suite-models/llama/test_llama.py +++ b/experimental/regression_suite/shark-test-suite-models/llama/test_llama.py @@ -16,11 +16,6 @@ ############################################################################### # Fixtures ############################################################################### -llama_real_weights = fetch_source_fixture( - "https://sharkblobs.blob.core.windows.net/halo-models/llm-dev/llama3_8b/8b_f16.irpa", - group="llama_fp16_8b", -) - llama_mlir = fetch_source_fixture( "https://sharkpublic.blob.core.windows.net/sharkpublic/halo-models/llm-dev/llama3_8b/8b_f16_decomposed_11_22.mlir", group="llama_fp16_8b",