File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
examples/pytorch/diffusion_model/diffusers/flux Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -36,12 +36,13 @@ function init_params {
3636
3737# run_tuning
3838function 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 \
You can’t perform that action at this time.
0 commit comments