Browse Source

Merge pull request #464 from jmorganca/mxyng/fix-num-keep

fix num_keep
Michael Yang 1 year ago
parent
commit
7efbc84320
1 changed files with 2 additions and 1 deletions
  1. 2 1
      server/routes.go

+ 2 - 1
server/routes.go

@@ -117,12 +117,13 @@ func load(ctx context.Context, model *Model, reqOpts map[string]interface{}, ses
 			if err != nil {
 				return err
 			}
+
 			tokensNoSystem, err := llmModel.Encode(ctx, promptNoSystem)
 			if err != nil {
 				return err
 			}
 
-			opts.NumKeep = len(tokensWithSystem) - len(tokensNoSystem) + 1
+			opts.NumKeep = len(tokensWithSystem) - len(tokensNoSystem)
 
 			llmModel.SetOptions(opts)
 		}