vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-11 07:44:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #778 from Conless/fix-shape

Browse source 
[Fix] Shape error in KLinearMarlin
This commit is contained in:
Atream 2025-03-03 19:58:52 +08:00 • committed by GitHub
commit f46b3fd51c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 0 additions and 1 deletions
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):