File tree Expand file tree Collapse file tree 1 file changed +21
-1
lines changed
Expand file tree Collapse file tree 1 file changed +21
-1
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments