From 80498f76de83bdf6ca59f1e9442dad9331168179 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Wed, 23 Apr 2025 16:22:22 -0700 Subject: [PATCH] fix build --- model/models/qwen25vl/model.go | 1 + model/models/qwen25vl/model_vision.go | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/model/models/qwen25vl/model.go b/model/models/qwen25vl/model.go index 1f30187ff..b8894616d 100644 --- a/model/models/qwen25vl/model.go +++ b/model/models/qwen25vl/model.go @@ -164,5 +164,6 @@ func (m *Model) Forward(ctx ml.Context, batch input.Batch) (ml.Tensor, error) { func init() { model.Register("qwen25vl", New) + model.Register("qwen2", New) model.Register("qwen2vl", New) } diff --git a/model/models/qwen25vl/model_vision.go b/model/models/qwen25vl/model_vision.go index da3f1b7a0..b86026181 100644 --- a/model/models/qwen25vl/model_vision.go +++ b/model/models/qwen25vl/model_vision.go @@ -303,7 +303,7 @@ func (m *VisionModel) Forward(ctx ml.Context, pixelValues ml.Tensor, grid *Grid) m.patchSize, // patch size, e.g., 14 ) - rope := rope(ctx, grid) + rope(ctx, grid) // spatialMergeSize := 2 // TODO: get this from config // // Create the position IDs tensor with correct dimensions