Skip to content

Commit e9a200f

Browse files
committed
undo high precision hadamard
Signed-off-by: Kyle Sayers <[email protected]>
1 parent bd4a991 commit e9a200f

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

src/compressed_tensors/transform/factory/hadamard.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,6 @@ def forward(self, value: Tensor) -> Tensor:
104104
if self.args.inverse:
105105
weight = weight.T
106106

107-
tmp = apply_transform_weight(
107+
return apply_transform_weight(
108108
weight, value, self.args.location, self.module_type
109-
)
110-
111-
112-
return (tmp.to(torch.float64) / self._scale).to(tmp.dtype)
109+
) / self._scale

0 commit comments

Comments
 (0)