Fixed filename bug

realfake/bin/generate_images.py

@@ -33,10 +33,12 @@ def main(args: GenerateImagesParams) -> None:
         (prompt_dir/"prompt.txt").write_text(prompt)
 
         left_images = args.num_images
+        count = 0
         while left_images > 0:
             output = pipe(prompt, num_images_per_prompt=min(batch_size, left_images))            
-            for i, image in enumerate(output.images):
-                image.save(prompt_dir/f"{i}.png")
+            for image in output.images:
+                image.save(prompt_dir/f"{count}.png")
+                count += 1
             left_images -= batch_size