update yml and remove the weight fixture until required

Signed-off-by: dan <[email protected]>
iree-org · Nov 26, 2024 · 6e305dc · 6e305dc
1 parent 98cdd22
commit 6e305dc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git 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
@@ -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",