diff --git a/utils/neuralmagic/sparsification_manager.py b/utils/neuralmagic/sparsification_manager.py index a79c79c64210..1e872cdac5e7 100644 --- a/utils/neuralmagic/sparsification_manager.py +++ b/utils/neuralmagic/sparsification_manager.py @@ -4,7 +4,7 @@ import torch from sparseml.pytorch.optim import ScheduledModifierManager -from sparseml.pytorch.utils import SparsificationGroupLogger +from sparseml.core.logger import SparsificationGroupLogger from utils.loggers import Loggers from utils.loss import ComputeLoss