Add missing parameters

Author: 67372a

flux_train_network.py

@@ -352,6 +352,8 @@ def get_noise_pred_and_target(
         train_unet,
         fixed_timesteps=None,
         train=True,
+        min_timestep_override=None,
+        max_timestep_override=None,
 ):      
         dtype_to_use = torch.float64 if args.loss_related_use_float64 else torch.float32
         with torch.autocast(dtype=dtype_to_use, device_type=str(accelerator.device)):