DPM++2MKarras

2024-03-05 02:04:44 -06:00
parent e3caf01a40
commit ad14a1647d
1 changed files with 1 additions and 0 deletions
--- a/lib/replicate.go
+++ b/lib/replicate.go
@ -72,6 +72,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
 		"width":                  1024,
 		"height":                 1024,
 		"prompt":                 prompt,
+		"scheduler":              "DPM++2MKarras",
 		"num_outputs":            1,
 		"num_inference_steps":    60,
 		"apply_watermark":        false,