>>102596217
--fp8_base --highvram --timestep_sampling sigma --model_prediction_type raw --discrete_flow_shift 1.0 --loss_type l2 --cache_text_encoder_outputs
Try this additional parameters. Failing that, grab a flux training config on this thread.
https://github.com/bmaltais/kohya_ss/issues/2701