server: accept slg_scale

stduhpf · stduhpf · commit f87d5f24ed02 · 2024-11-04T19:27:30.000+01:00
diff --git a/examples/server/main.cpp b/examples/server/main.cpp
@@ -778,6 +778,12 @@ void parseJsonPrompt(std::string json_str, SDParams* params) {
         params->input_id_images_path = input_id_images_path;
     } catch (...) {
     }
+    try {
+        std::string slg_scale = payload["slg_scale"];
+        params->slg_scale = stof(slg_scale);
+    } catch (...) {
+    }
+    // TODO: more slg settings (layers, start and end)
 }
 
 // https://github.com/comfyanonymous/ComfyUI/blob/master/comfy/latent_formats.py#L152-L169

Original file line number	Diff line number	Diff line change
`@@ -778,6 +778,12 @@ void parseJsonPrompt(std::string json_str, SDParams* params) {`
`778`	`778`	`params->input_id_images_path = input_id_images_path;`
`779`	`779`	`} catch (...) {`
`780`	`780`	`}`
	`781`	`+ try {`
	`782`	`+ std::string slg_scale = payload["slg_scale"];`
	`783`	`+ params->slg_scale = stof(slg_scale);`
	`784`	`+ } catch (...) {`
	`785`	`+ }`
	`786`	`+ // TODO: more slg settings (layers, start and end)`
`781`	`787`	`}`
`782`	`788`
`783`	`789`	`// https://github.com/comfyanonymous/ComfyUI/blob/master/comfy/latent_formats.py#L152-L169`