format code

Author: leejet

mmdit.hpp

@@ -920,7 +920,7 @@ struct MMDiTRunner : public GGMLRunner {
                  struct ggml_tensor* y,
                  struct ggml_tensor** output     = NULL,
                  struct ggml_context* output_ctx = NULL,
-                 std::vector<int> skip_layers = std::vector<int>()) {
+                 std::vector<int> skip_layers    = std::vector<int>()) {
         // x: [N, in_channels, h, w]
         // timesteps: [N, ]
         // context: [N, max_position, hidden_size]([N, 154, 4096]) or [1, max_position, hidden_size]

stable-diffusion.cpp

@@ -789,11 +789,10 @@ class StableDiffusionGGML {
         bool has_unconditioned = cfg_scale != 1.0 && uncond.c_crossattn != NULL;
         bool has_skiplayer     = slg_scale != 0.0 && skip_layers.size() > 0;
 
-
         // denoise wrapper
         struct ggml_tensor* out_cond   = ggml_dup_tensor(work_ctx, x);
         struct ggml_tensor* out_uncond = NULL;
-        struct ggml_tensor* out_skip = NULL;
+        struct ggml_tensor* out_skip   = NULL;
 
         if (has_unconditioned) {
             out_uncond = ggml_dup_tensor(work_ctx, x);
@@ -887,8 +886,8 @@ class StableDiffusionGGML {
                 negative_data = (float*)out_uncond->data;
             }
 
-            int stepCount          = sigmas.size();
-            bool is_skiplayer_step = has_skiplayer && step > (int)(skip_layer_start * stepCount) && step < (int)(skip_layer_end * stepCount);
+            int step_count         = sigmas.size();
+            bool is_skiplayer_step = has_skiplayer && step > (int)(skip_layer_start * step_count) && step < (int)(skip_layer_end * step_count);
             float* skip_layer_data = NULL;
             if (is_skiplayer_step) {
                 LOG_DEBUG("Skipping layers at step %d\n", step);

stable-diffusion.h

@@ -163,10 +163,10 @@ SD_API sd_image_t* txt2img(sd_ctx_t* sd_ctx,
                            float style_strength,
                            bool normalize_input,
                            const char* input_id_images_path,
-                           std::vector<int> skip_layers  = {},
-                           float slg_scale               = 2.5,
-                           float skip_layer_start        = 0.01,
-                           float skip_layer_end          = 0.2);
+                           std::vector<int> skip_layers = {},
+                           float slg_scale              = 2.5,
+                           float skip_layer_start       = 0.01,
+                           float skip_layer_end         = 0.2);
 
 SD_API sd_image_t* img2img(sd_ctx_t* sd_ctx,
                            sd_image_t init_image,