fix quality errors

neuralmagic · Oct 25, 2023 · b963683 · b963683
1 parent a934051
commit b963683
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/src/sparseml/pytorch/recipe_template/main.py b/src/sparseml/pytorch/recipe_template/main.py
@@ -31,7 +31,6 @@
     MFACPruningModifier,
     MovementPruningModifier,
     OBSPruningModifier,
-    PowerpropagationModifier,
 )
 from sparseml.pytorch.sparsification.quantization.legacy_modifier_quantization import (
     QuantizationModifier,

diff --git a/src/sparseml/pytorch/sparsification/pruning/modifier_powerpropagation.py b/src/sparseml/pytorch/sparsification/pruning/modifier_powerpropagation.py
@@ -35,7 +35,6 @@
     NamedLayerParam,
     get_named_layers_and_params_by_regex,
     get_prunable_layers,
-    tensors_to_device,
 )
 from sparseml.pytorch.utils.logger import LoggerManager
 from sparseml.utils import ALL_PRUNABLE_TOKEN, ALL_TOKEN, validate_str_iterable