XYZ if not Include Sub Grids do not save Sub Grid

k1nos · Sep 16, 2023 · d2878a8 · d2878a8
1 parent 102b661
commit d2878a8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
@@ -773,6 +773,8 @@ def cell(x, y, z, ix, iy, iz):
                 # TODO: See previous comment about intentional data misalignment.
                 adj_g = g-1 if g > 0 else g
                 images.save_image(processed.images[g], p.outpath_grids, "xyz_grid", info=processed.infotexts[g], extension=opts.grid_format, prompt=processed.all_prompts[adj_g], seed=processed.all_seeds[adj_g], grid=True, p=processed)
+                if not include_sub_grids:  # if not include_sub_grids then skip saving after the first grid
+                    break
 
         if not include_sub_grids:
             # Done with sub-grids, drop all related information: