Skip to content

Commit 740a1c2

Browse files
committed
tests python command
1 parent 4eac1e0 commit 740a1c2

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

.github/workflows/ci_sdxl_flux_serving.yml

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ jobs:
4141
VENV_DIR: ${{ github.workspace }}/.venv
4242
HF_HOME: "/amdshark-dev/data/huggingface"
4343
HF_TOKEN: ${{ secrets.HF_FLUX_TOKEN }}
44+
HIP_VISIBLE_DEVICES: 0
4445
steps:
4546
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
4647

@@ -60,9 +61,21 @@ jobs:
6061
- name: Run SDXL (export)
6162
run: |
6263
mkdir -p output_artifacts/sdxl
64+
sudo mkdir -p /amdshark-dev/data/huggingface
65+
cd shortfin
66+
python -m shortfin_apps.sd.server \
67+
--device=hip \
68+
--device_ids=0 \
69+
--build_preference=export \
70+
--port 8112 --force_update=True |& tee output_artifacts/sdxl/sdxl_export.log
71+
72+
- name: Run SDXL (bash)
73+
run: |
74+
mkdir -p output_artifacts/sdxl
75+
sudo mkdir -p /amdshark-dev/data/huggingface
6376
bash scripts/run_sdxl_flux.sh \
6477
--model sdxl \
65-
--build_preference export --force-update 2>&1 | tee output_artifacts/sdxl/sdxl_precompile.log
78+
--build_preference export --force-update |& tee output_artifacts/sdxl/sdxl_precompile.log
6679
6780
- name: Run Flux-Dev (precompiled)
6881
if: always()

0 commit comments

Comments
 (0)