server: accept slg_scale

stduhpf · Nov 4, 2024 · f87d5f2 · f87d5f2
1 parent 8577446
commit f87d5f2
Showing 1 changed file with 6 additions and 0 deletions.
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