Update app.py

app.py

@@ -53,8 +53,8 @@ If you have any questions, please feel free to open an issue or directly reach u
 
 
 model_num_of_layers = {
-    'Stable Diffusion 1.4': 12,
-    'Stable Diffusion 2.1': 22,
+    'Stable Diffusion 1.4': 13,
+    'Stable Diffusion 2.1': 23,
 }
 
 # def run_for_examples(prompt, model, seed, skip):
@@ -67,7 +67,9 @@ def generate_images(prompt, model, seed, skip):
     gr.Info('Generating images from intermediate layers..')
     all_images = []  # Initialize a list to store all images
     max_num_of_layers = model_num_of_layers[model]
-    for skip_layers in range(max_num_of_layers - 1, -1, -1 * skip):
+    start_layer = max_num_of_layers % skip
+    
+    for skip_layers in range(max_num_of_layers - start_layer, -1, -1 * skip):
         # Pass the model and seed to the get_images function
         images = get_images(prompt, skip_layers=skip_layers, model=model, seed=seed)
         all_images.append((images[0], f'layer_{max_num_of_layers - skip_layers}'))