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

Commit 4cdbb8d

Browse files
committed
self reference to H
1 parent 57cb5e9 commit 4cdbb8d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sparseml/modifiers/quantization/gptq/utils/gptq_wrapper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,9 +111,9 @@ def fasterprune(
111111
W[:, dead] = 0
112112

113113
if actorder:
114-
perm = torch.argsort(torch.diag(H), descending=True)
114+
perm = torch.argsort(torch.diag(self.H), descending=True)
115115
W = W[:, perm]
116-
H = H[perm][:, perm]
116+
self.H = self.H[perm][:, perm]
117117
invperm = torch.argsort(perm)
118118

119119
Losses = torch.zeros(self.rows, device=self.dev)

0 commit comments

Comments
 (0)