github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

small fixes

Browse Source
This commit is contained in:
Ean Garvey
2024-06-02 11:28:37 -05:00
parent 349e9f70fb
commit 9163c1fc50
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
apps/shark_studio/api/sd.py
View File

@@ -254,7 +254,7 @@ class StableDiffusion:
vmfbs, weights = self.sd_pipe.check_prepared(
mlirs, vmfbs, weights, interactive=False
)
progress(.5, desc=f"Artifacts ready!")
progress(0.5, desc=f"Artifacts ready!")
progress(0.75, desc=f"Loading models and weights...")
self.sd_pipe.load_pipeline(
@@ -297,6 +297,8 @@ def shark_sd_fn_dict_input(sd_kwargs: dict, *, progress=gr.Progress()):
sd_kwargs[key] = None
if sd_kwargs[key] in ["None"]:
sd_kwargs[key] = ""
if key in ["steps", "height", "width", "batch_count", "batch_size"]:
sd_kwargs[key] = int(sd_kwargs[key])
if key == "seed":
sd_kwargs[key] = int(sd_kwargs[key])
@@ -487,7 +489,7 @@ def shark_sd_fn(
return (generated_imgs, "")
def get_next_seed(seed, seed_increment):
def get_next_seed(seed, seed_increment: str | int = 1):
if isinstance(seed_increment, int):
return int(seed + seed_increment)
elif seed_increment == "random":