• Aubrey Li's avatar
    Fix TypeError when invoke KLinearCPUInfer.forward() · 12a4c631
    Aubrey Li authored
    Fix the following error:
    
      File "/home/aubrey/work/ktransformers/ktransformers/operators/linear.py", line 825, in forward
        y = self.generate_linear.forward(x, bsz_tensor)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
    TypeError: KLinearCPUInfer.forward() takes 2 positional arguments but 3 were given
    12a4c631
linear.py 33.2 KB