Skip to content

Commit 72c8d96

Browse files
committed
server: add more slg params
1 parent c6356c7 commit 72c8d96

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

examples/server/main.cpp

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -780,10 +780,30 @@ void parseJsonPrompt(std::string json_str, SDParams* params) {
780780
}
781781
try {
782782
std::string slg_scale = payload["slg_scale"];
783-
params->slg_scale = stof(slg_scale);
783+
params->slg_scale = stof(slg_scale);
784784
} catch (...) {
785785
}
786786
// TODO: more slg settings (layers, start and end)
787+
try {
788+
std::vector<int> skip_layers = payload["skip_layers"].get<std::vector<int>>();
789+
params->skip_layers.clear();
790+
for (int i = 0; i < skip_layers.size(); i++) {
791+
params->skip_layers.push_back(skip_layers[i]);
792+
}
793+
} catch (...) {
794+
}
795+
try {
796+
// skip_layer_start
797+
float skip_layer_start = payload["skip_layer_start"].get<float>();
798+
params->skip_layer_start = skip_layer_start;
799+
} catch (...) {
800+
}
801+
try {
802+
// skip_layer_end
803+
float skip_layer_end = payload["skip_layer_end"].get<float>();
804+
params->skip_layer_end = skip_layer_end;
805+
} catch (...) {
806+
}
787807
}
788808

789809
// https://github.com/comfyanonymous/ComfyUI/blob/master/comfy/latent_formats.py#L152-L169

0 commit comments

Comments
 (0)