Skip to content

Commit 8564f48

Browse files
authored
update flux scripts (#2351)
Signed-off-by: chensuyue <suyue.chen@intel.com>
1 parent f88de2e commit 8564f48

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/pytorch/diffusion_model/diffusers/flux/run_quant.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,13 @@ function init_params {
3636

3737
# run_tuning
3838
function run_tuning {
39+
dataset_location=${dataset_location:="captions_source.tsv"}
3940
tuned_checkpoint=${tuned_checkpoint:="saved_results"}
4041

4142
if [ "${topology}" = "flux_fp8" ]; then
42-
extra_cmd="--scheme FP8 --iters 0 --dataset captions_source.tsv --quantize"
43+
extra_cmd="--scheme FP8 --iters 0 --dataset ${dataset_location} --quantize"
4344
elif [ "${topology}" = "flux_mxfp8" ]; then
44-
extra_cmd="--scheme MXFP8 --iters 1000 --dataset captions_source.tsv --quantize"
45+
extra_cmd="--scheme MXFP8 --iters 1000 --dataset ${dataset_location} --quantize"
4546
fi
4647

4748
python3 main.py \

0 commit comments

Comments
 (0)