highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-04-01 00:19:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

gemma2: Remove second call to Rows

Browse Source 
Looks like a merge conflict that broke the model.
This commit is contained in:
Jesse Gross 2025-03-19 15:34:41 -07:00 committed by Jesse Gross
parent 2ddacd7516
commit b078dd157c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model/models/gemma2/model.go
View File

@ -211,8 +211,7 @@ func (m *Model) Forward(ctx ml.Context, opts input.Options) (ml.Tensor, error) {
// final logit softcap
hiddenState = hiddenState.Scale(ctx, 1.0/float64(m.Options.finalLogitSoftcap))
hiddenState = hiddenState.Tanh(ctx)
hiddenState = hiddenState.Scale(ctx, float64(m.Options.finalLogitSoftcap))
return hiddenState.Rows(ctx, outputs), nil
return hiddenState.Scale(ctx, float64(m.Options.finalLogitSoftcap)), nil
}
func init() {