jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix build

Browse Source
This commit is contained in:
Bruce MacDonald 2025-04-23 16:22:22 -07:00
parent f8b48aa784
commit 80498f76de
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model/models/qwen25vl/model.go
View File

@ -164,5 +164,6 @@ func (m *Model) Forward(ctx ml.Context, batch input.Batch) (ml.Tensor, error) {
func init() { func init() {
model.Register("qwen25vl", New) model.Register("qwen25vl", New)
model.Register("qwen2", New)
model.Register("qwen2vl", New) model.Register("qwen2vl", New)
} }

2
model/models/qwen25vl/model_vision.go
View File

@ -303,7 +303,7 @@ func (m *VisionModel) Forward(ctx ml.Context, pixelValues ml.Tensor, grid *Grid)
m.patchSize, // patch size, e.g., 14 m.patchSize, // patch size, e.g., 14
) )
rope := rope(ctx, grid) rope(ctx, grid)
// spatialMergeSize := 2 // TODO: get this from config // spatialMergeSize := 2 // TODO: get this from config
// // Create the position IDs tensor with correct dimensions // // Create the position IDs tensor with correct dimensions