From c088247aa1cd545d38d12bb88769fe876fd3cdb3 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Fri, 31 May 2024 14:26:17 -0500 Subject: [PATCH] Fix default configs, config loading, and add warnings/early returns for bad configs. --- apps/shark_studio/api/sd.py | 4 ---- apps/shark_studio/web/utils/default_configs.py | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py index de2617f8..70e97199 100644 --- a/apps/shark_studio/api/sd.py +++ b/apps/shark_studio/api/sd.py @@ -262,8 +262,6 @@ class StableDiffusion: control_mode, hints, ): - if seed == -1: - seed = randint(0, sys.maxsize) img = self.sd_pipe.generate_images( prompt, negative_prompt, @@ -458,8 +456,6 @@ def shark_sd_fn( sd_kwargs, ) generated_imgs.extend(out_imgs) - # TODO: make seed changes over batch counts more configurable. - submit_run_kwargs["seed"] = submit_run_kwargs["seed"] + 1 yield generated_imgs, status_label( "Stable Diffusion", current_batch + 1, batch_count, batch_size ) diff --git a/apps/shark_studio/web/utils/default_configs.py b/apps/shark_studio/web/utils/default_configs.py index aef62e1b..b52ba63f 100644 --- a/apps/shark_studio/web/utils/default_configs.py +++ b/apps/shark_studio/web/utils/default_configs.py @@ -92,4 +92,4 @@ default_sd_configs = { "default_sd_config.json": default_sd_config, "sdxl-30steps.json": sdxl_30steps, "sdxl-turbo.json": sdxl_turbo, -} +} \ No newline at end of file