Skip to content

GitLab

Unverified Commit f46b3fd5 authored by Atream's avatar Atream Committed by GitHub
Browse files

Merge pull request #778 from Conless/fix-shape 

[Fix] Shape error in KLinearMarlin
parents 4e43e8a4 01755a60
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
ktransformers/operators/linear.py
View file @ f46b3fd5
...@@ -347,7 +347,6 @@ class KLinearMarlin(KLinearBase): ...@@ -347,7 +347,6 @@ class KLinearMarlin(KLinearBase):
orig_shape[-1] = self.out_features orig_shape[-1] = self.out_features
if self.has_bias: if self.has_bias:
x = x + self.bias x = x + self.bias
orig_shape[-1] = self.n
return x.reshape(orig_shape).to(orig_dtype) return x.reshape(orig_shape).to(orig_dtype)
def unload(self): def unload(self):
......
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