Use top-50 as the default for gpt

pcuenca · pcuenca · commit 209f0c5005dd · 2023-05-18T17:16:37.000+02:00
diff --git a/Sources/Models/LanguageModel.swift b/Sources/Models/LanguageModel.swift
@@ -133,7 +133,7 @@ extension LanguageModel: TextGenerationModel {
         switch modelName.lowercased() {
         case let x where x.contains("gpt"):
             config.doSample = true
-            config.topK = 10
+            config.topK = 50
         default: break
         }
         return config

Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,7 @@ extension LanguageModel: TextGenerationModel {`
`133`	`133`	`switch modelName.lowercased() {`
`134`	`134`	`case let x where x.contains("gpt"):`
`135`	`135`	`config.doSample = true`
`136`		`- config.topK = 10`
	`136`	`+ config.topK = 50`
`137`	`137`	`default: break`
`138`	`138`	`}`
`139`	`139`	`return config`