From a4a33f7f9bead054aadb517becfb725dc3a3ca90 Mon Sep 17 00:00:00 2001 From: chensuyue Date: Sun, 7 Dec 2025 15:22:50 +0800 Subject: [PATCH] update flux scripts Signed-off-by: chensuyue --- examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh b/examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh index d13c3bcf470..8b72db4d77c 100644 --- a/examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh +++ b/examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh @@ -36,12 +36,13 @@ function init_params { # run_tuning function run_tuning { + dataset_location=${dataset_location:="captions_source.tsv"} tuned_checkpoint=${tuned_checkpoint:="saved_results"} if [ "${topology}" = "flux_fp8" ]; then - extra_cmd="--scheme FP8 --iters 0 --dataset captions_source.tsv --quantize" + extra_cmd="--scheme FP8 --iters 0 --dataset ${dataset_location} --quantize" elif [ "${topology}" = "flux_mxfp8" ]; then - extra_cmd="--scheme MXFP8 --iters 1000 --dataset captions_source.tsv --quantize" + extra_cmd="--scheme MXFP8 --iters 1000 --dataset ${dataset_location} --quantize" fi python3 main.py \