test sdxl ci

Signed-off-by: Vivek Agrawal <197589114+amd-vivekag@users.noreply.github.com>

Author: amd-vivekag

.github/workflows/ci_sdxl_flux_serving.yml

@@ -8,6 +8,9 @@ name: Release SDXL and FLUX Shortfin Serving Tests
 
 on:
   workflow_dispatch:
+  push:
+  branches:
+    - branch-name
   schedule:
     # Weekdays at 11:00 AM UTC = 03:00 AM PST / 04:00 AM PDT
     - cron: "0 9 * * *"
@@ -38,7 +41,7 @@ jobs:
         shell: bash
     env:
       VENV_DIR: ${{ github.workspace }}/.venv
-      HF_HOME: "/amdshark-cache/data/huggingface"
+      HF_HOME: "/amdshark-dev/data/huggingface"
       HF_TOKEN: ${{ secrets.HF_FLUX_TOKEN }}
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
@@ -56,15 +59,17 @@ jobs:
       - name: Install pip deps
         run: bash scripts/setenv.sh --nightly
 
-      - name: Run SDXL (precompiled)
+      - name: Run SDXL (export)
         run: |
+          mkdir -p output_artifacts/sdxl
           bash scripts/run_sdxl_flux.sh \
             --model sdxl \
-            --build_preference precompiled 2>&1 | tee output_artifacts/sdxl/sdxl_precompile.log
+            --build_preference export --force-update 2>&1 | tee output_artifacts/sdxl/sdxl_precompile.log
 
       - name: Run Flux-Dev (precompiled)
         if: always()
         run: |
+          mkdir -p output_artifacts/flux_dev
           bash scripts/run_sdxl_flux.sh \
             --model flux_dev \
             --build_preference precompiled \
@@ -73,6 +78,7 @@ jobs:
       - name: Run Flux-Dev (compile)
         if: always()
         run: |
+          mkdir -p output_artifacts/flux_dev
           bash scripts/run_sdxl_flux.sh \
             --model flux_dev \
             --build_preference compile \
@@ -81,6 +87,7 @@ jobs:
       - name: Run Flux-Schnell (precompiled)
         if: always()
         run: |
+          mkdir -p output_artifacts/flux_schnell
           bash scripts/run_sdxl_flux.sh \
             --model flux_schnell \
             --build_preference precompiled \
@@ -89,6 +96,7 @@ jobs:
       - name: Run Flux-Schnell (compile)
         if: always()
         run: |
+          mkdir -p output_artifacts/flux_schnell
           bash scripts/run_sdxl_flux.sh \
             --model flux_schnell \
             --build_preference compile \

scripts/run_sdxl_flux.sh

@@ -2,6 +2,7 @@
 
 export BUILD_PREFERENCE="precompiled"
 export PORT=8112
+export FORCE_UPDATE=False
 SCRIPT_DIR=$(dirname $(realpath "$0"))
 SHORTFIN_SRC=$SCRIPT_DIR/../shortfin
 HF_HOME_DIR=${HF_HOME:-"$HOME/.cache/huggingface"}
@@ -15,7 +16,7 @@ function run_sdxl_model() {
         --device=hip \
         --device_ids=0 \
         --build_preference=$BUILD_PREFERENCE \
-        --port $PORT &
+        --port $PORT --force_update=$FORCE_UPDATE &
 
     SHORTFIN_PROCESS=$!
     wait_for_server $PORT
@@ -112,6 +113,10 @@ while [[ "$1" != "" ]]; do
         shift
         export HF_TOKEN=$1
         ;;
+    --force-update)
+        shift
+        export FORCE_UPDATE=True
+        ;;
     -h | --help)
         echo "Usage: $0 [--<different flags>] "
         echo "--build_preference : Preference for builder artifact generation, default: compile"