vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-06 04:30:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix: wrong shape in KLinearMarlin.

Browse source
This commit is contained in:
Yi Pan 2025-03-03 17:34:37 +08:00
parent 4e43e8a4ee
commit 01755a60c0
No known key found for this signature in database
GPG key ID: DEBAB8AF77793700
1 changed files with 0 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
ktransformers/operators/linear.py
View file

@ -347,7 +347,6 @@ class KLinearMarlin(KLinearBase):
orig_shape[-1] = self.out_features
if self.has_bias:
x = x + self.bias
orig_shape[-1] = self.n
return x.reshape(orig_shape).to(orig_dtype)
def unload(self):