github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

More noticeable seed changes

Browse Source
This commit is contained in:
Ean Garvey
2024-06-02 11:39:21 -05:00
parent 36b8c2fd6d
commit 9957c96014
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -461,6 +461,7 @@ def shark_sd_fn(
generated_imgs = []
if seed in [-1, "-1"]:
seed = randint(0, 4294967295)
seed_increment = "random"
print(f"\n[LOG] Random seed: {seed}")
progress(None, desc=f"Generating...")
@@ -489,11 +490,14 @@ def shark_sd_fn(
return (generated_imgs, "")
def get_next_seed(seed, seed_increment: str | int = 1):
def get_next_seed(seed, seed_increment: str | int = 10):
if isinstance(seed_increment, int):
print(f"\n[LOG] Seed after batch increment: {seed + seed_increment}")
return int(seed + seed_increment)
elif seed_increment == "random":
return randint(0, sys.maxsize)
seed = randint(0, 4294967295)
print(f"\n[LOG] Random seed: {seed}")
return seed
def unload_sd():

2
apps/shark_studio/modules/schedulers.py
View File

@@ -101,7 +101,7 @@ def export_scheduler_model(model):
scheduler_model_map = {
#"PNDM": export_scheduler_model("PNDMScheduler"),
"PNDM": export_scheduler_model("PNDMScheduler"),
# "DPMSolverSDE": export_scheduler_model("DpmSolverSDEScheduler"),
"EulerDiscrete": export_scheduler_model("EulerDiscreteScheduler"),
"EulerAncestralDiscrete": export_scheduler_model("EulerAncestralDiscreteScheduler"),