Added lorm. WIP

2026-03-13 06:29:48 +00:00 · 2023-10-26 18:23:51 -06:00
parent 9636194c09
commit 0a79ac9604
3 changed files with 441 additions and 4 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -456,7 +456,7 @@ class StableDiffusion:
                            **extra
                        ).images[0]

-                    gen_config.save_image(img)
+                    gen_config.save_image(img, i)

        # clear pipeline and cache to reduce vram usage
        del pipeline