Style

neuralmagic · Oct 15, 2023 · cedbd8f · cedbd8f
1 parent bdd9939
commit cedbd8f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/sparseml/transformers/sparsification/trainer.py b/src/sparseml/transformers/sparsification/trainer.py
@@ -41,11 +41,10 @@
 
 from sparseml.core.logger import LoggerManager, TensorBoardLogger, WANDBLogger
 from sparseml.pytorch.optim import ScheduledModifierManager, ScheduledOptimizer
-from sparseml.pytorch.utils import ModuleSparsificationInfo
-
 from sparseml.pytorch.sparsification.quantization.helpers import (
     initialize_channel_wise_scale_zp,
 )
+from sparseml.pytorch.utils import ModuleSparsificationInfo
 from sparseml.transformers.utils import SparseAutoModel
 from sparseml.transformers.utils.helpers import RECIPE_NAME