model: add lfm2 architecture and LFM2.5-1.2B-Thinking support (#13792)

Co-Authored-By: TommyBoiss <165361500+TommyBoiss@users.noreply.github.com>
This commit is contained in:
Jeffrey Morgan
2026-01-20 12:20:53 -08:00
committed by GitHub
parent 31085d5e53
commit 01cf7445f3
18 changed files with 3441 additions and 2 deletions

View File

@@ -162,6 +162,7 @@ type Tensor interface {
AvgPool2D(ctx Context, k, s int, p float32) Tensor
Conv2D(ctx Context, weight Tensor, s0, s1, p0, p1, d0, d1 int) Tensor
Conv3D(ctx Context, weight Tensor, c, s0, s1, s2, p0, p1, p2, d0, d1, d2 int) Tensor
SSMConv(ctx Context, kernel Tensor) Tensor
IM2Col(ctx Context, weight Tensor, s0, s1, p0, p1, d0, d1 int) Tensor