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