Add generation params to DB, fix GUI star layout/refresh, add sample generator script

2026-06-28 16:56:44 +03:00
parent 68cba874d5
commit 09f6f377c7
4 changed files with 223 additions and 60 deletions
@@ -282,6 +282,15 @@ def generate_sprite(
            "steps": steps,
            "remove_bg": remove_bg,
            "pixel_size": pixel_size,
+            "model": MODEL_DIR,
+            "lora": LORA_DIR,
+            "lcm": LCM_LORA_DIR,
+            "lora_scale": PIXEL_LORA_SCALE,
+            "lcm_scale": LCM_LORA_SCALE,
+            "negative_prompt": NEGATIVE_PROMPT,
+            "guidance_scale": 1.5,
+            "scheduler": "LCMScheduler",
+            "full_prompt": full_prompt,
        },
        image_path=output_path,
    )
@@ -361,6 +370,15 @@ def batch_generate(
                "steps": steps,
                "remove_bg": remove_bg,
                "pixel_size": pixel_size,
+                "model": MODEL_DIR,
+                "lora": LORA_DIR,
+                "lcm": LCM_LORA_DIR,
+                "lora_scale": PIXEL_LORA_SCALE,
+                "lcm_scale": LCM_LORA_SCALE,
+                "negative_prompt": NEGATIVE_PROMPT,
+                "guidance_scale": 1.5,
+                "scheduler": "LCMScheduler",
+                "full_prompt": full_prompt,
            },
            image_path=output_path,
        )