diff --git a/src/sparseml/pytorch/torchvision/train.py b/src/sparseml/pytorch/torchvision/train.py
index 747348c7f1d..101c08ae842 100644
--- a/src/sparseml/pytorch/torchvision/train.py
+++ b/src/sparseml/pytorch/torchvision/train.py
@@ -772,7 +772,6 @@ def _create_model(
                 num_classes=num_classes,
             )
 
-
         if isinstance(model, tuple):
             model, arch_key = model
     elif arch_key in torchvision.models.__dict__: