Skip to content

GitLab

Unverified Commit d9dc8e56 authored by Casper's avatar Casper Committed by GitHub
Browse files

Fix double bias (#383)

parent 68c727a1
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
awq/modules/linear/gemm.py
View file @ d9dc8e56
......@@ -253,7 +253,6 @@ class WQLinear_GEMM(nn.Module):
if input_dtype != torch.float16:
out = out.to(dtype=input_dtype)
out = out + self.bias if self.bias is not None else out
return out.reshape(out_shape)
def extra_repr(self) -> str:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment