Skip to content

Commit

Permalink
Update pytorch.py
Browse files Browse the repository at this point in the history
removing stray change
  • Loading branch information
robertgshaw2-neuralmagic authored Oct 30, 2023
1 parent 6e99401 commit 12831da
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/sparseml/modifiers/obcq/pytorch.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,6 @@ def initialize_obcq(
:param model: PyTorch model to sparsify
:param device: device to run sparsification on, preferably a GPU
"""

self.model = model
self.compressible_layers_ = self.compressible_layers()
self.model = self.model.model
Expand Down

0 comments on commit 12831da

Please sign in to comment.