From cf37dc2154bce8c774615a8da3051064727b87fd Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Fri, 4 Mar 2022 13:57:19 -0500 Subject: [PATCH] update modifiers import for latest sparseml refactors --- src/transformers/sparse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/sparse.py b/src/transformers/sparse.py index 633e963aaf28..9fd6e528cc80 100644 --- a/src/transformers/sparse.py +++ b/src/transformers/sparse.py @@ -9,10 +9,10 @@ import torch import onnxruntime -from sparseml.pytorch.optim import LayerPruningModifier, QuantizationModifier from sparseml.pytorch.optim.manager import ScheduledModifierManager from sparseml.pytorch.optim.optimizer import ScheduledOptimizer from sparseml.pytorch.utils import logger +from sparseml.pytorch.sparsification import LayerPruningModifier, QuantizationModifier from transformers import Trainer from transformers.file_utils import RECIPE_NAME, WEIGHTS_NAME