Upload test_grid.py with huggingface_hub

test_grid.py

@@ -25,6 +25,12 @@ PROMPT_PRESETS = {
         "soft fur shading, warm colors, subtle highlights, thin brown outline, "
         "cute face, dot eyes with highlights, pink blush, white background"
     ),
+    "reference": (
+        "cute plush toy, high resolution pixel art character, large visible pixels, "
+        "rounded chibi body, front-facing full body, warm soft shading, "
+        "brown pixel outline, glossy dot eyes, tiny nose, small smile, "
+        "pink cheek blush, clean white background"
+    ),
 }
 
 
@@ -78,6 +84,7 @@ def main():
     parser.add_argument("--prompt-presets", default="default,soft,sprite")
     parser.add_argument("--seed", type=int, default=42)
     parser.add_argument("--limit", type=int, default=0, help="Limit number of input images. 0 means all.")
+    parser.add_argument("--save-individual", action="store_true", help="Also save every generated image.")
     args = parser.parse_args()
 
     images = collect_images(args.input)
@@ -130,13 +137,16 @@ def main():
                 f"_st{strength:.2f}_cn{controlnet_scale:.2f}_g{guidance_scale:.1f}.png"
             )
             output_path = image_output_dir / filename
-            result["image"].save(output_path)
+            if args.save_individual:
+                result["image"].save(output_path)
+                print(f"saved {output_path}")
 
             title = f"{prompt_key}\nst={strength:.2f} cn={controlnet_scale:.2f} g={guidance_scale:.1f}"
             grid_items.append((title, result["image"]))
-            print(f"saved {output_path}")
 
-        save_grid(grid_items, image_output_dir / f"{stem}_grid.png", columns=4)
+        grid_path = image_output_dir / f"{stem}_grid.png"
+        save_grid(grid_items, grid_path, columns=4)
+        print(f"saved grid {grid_path}")
 
     print(f"done: {output_dir}")