From 4caca754ca7332b4018f311b313bdb1a818f69e8 Mon Sep 17 00:00:00 2001 From: Tanner Kvarfordt Date: Fri, 14 Jul 2023 17:25:55 -0600 Subject: [PATCH] Switched recommended text generation model to a model that supports more than 1 return sequence --- text_generation.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/text_generation.go b/text_generation.go index b6e57ea..0555b06 100644 --- a/text_generation.go +++ b/text_generation.go @@ -6,7 +6,7 @@ import ( "fmt" ) -const RecommendedTextGenerationModel = "gpt2" +const RecommendedTextGenerationModel = "gpt2-large" type TextGenerationRequest struct { // (Required) a string to be generated from