Switched recommended text generation model to a model that supports m…

…ore than 1 return sequence
Kardbord · Jul 14, 2023 · 4caca75 · 4caca75
1 parent 16fc2a9
commit 4caca75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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