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