diff --git a/src/maxdiffusion/train.py b/src/maxdiffusion/train.py
index 1bfcc942..b206a9a9 100644
--- a/src/maxdiffusion/train.py
+++ b/src/maxdiffusion/train.py
@@ -26,7 +26,6 @@
 from maxdiffusion.train_utils import (
     validate_train_config,
 )
-
 from maxdiffusion.trainers.stable_diffusion_trainer import StableDiffusionTrainer