Skip to content
This repository was archived by the owner on Jun 4, 2025. It is now read-only.

Commit 54608c7

Browse files
authored
update modifiers import for latest sparseml refactors (#34)
1 parent 410f9db commit 54608c7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/sparse.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@
99
import torch
1010

1111
import onnxruntime
12-
from sparseml.pytorch.optim import LayerPruningModifier, QuantizationModifier
1312
from sparseml.pytorch.optim.manager import ScheduledModifierManager
1413
from sparseml.pytorch.optim.optimizer import ScheduledOptimizer
1514
from sparseml.pytorch.utils import logger
15+
from sparseml.pytorch.sparsification import LayerPruningModifier, QuantizationModifier
1616
from transformers import Trainer
1717
from transformers.file_utils import RECIPE_NAME, WEIGHTS_NAME
1818

0 commit comments

Comments
 (0)