+ Model Select

2023-05-04 16:51:05 +03:00
parent fee62d1918
commit 4992174f21
4 changed files with 42 additions and 4 deletions
--- a/lib/prodia/prodia.go
+++ b/lib/prodia/prodia.go
@@ -22,7 +22,7 @@ func getProdiaKey() string {
 	return os.Getenv("PRODIA_KEY")
 }

-func RequestGeneration(prompt string, negative string) (*rtypes.ProdiaGenerateResponse, error) {
+func RequestGeneration(prompt string, negative string, modelChoice rtypes.ProdiaModel) (*rtypes.ProdiaGenerateResponse, error) {
 	req := rtypes.ProdiaGenerateRequest{
 		Model:          rtypes.Deliberate_v2,
 		Prompt:         prompt,