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

Add support for command arguments to the WebUi (#1000)

Browse Source 
Co-authored-by: powderluv <powderluv@users.noreply.github.com>
This commit is contained in:
drumicube
2023-02-12 04:20:21 +01:00
committed by GitHub
parent f64e1fb926
commit 0464c7e558
5 changed files with 51 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
apps/stable_diffusion/scripts/inpaint.py
View File

@@ -310,8 +310,8 @@ if __name__ == "__main__":
args.use_tuned,
)
for run in range(args.runs):
if run > 0:
for current_batch in range(args.batch_count):
if current_batch > 0:
seed = -1
seed = utils.sanitize_seed(seed)

6
apps/stable_diffusion/scripts/txt2img.py
View File

@@ -294,8 +294,8 @@ if __name__ == "__main__":
args.use_tuned,
)
for run in range(args.runs):
if run > 0:
for current_batch in range(args.batch_count):
if current_batch > 0:
seed = -1
seed = utils.sanitize_seed(seed)
@@ -325,7 +325,7 @@ if __name__ == "__main__":
text_output += (
f", batch size={args.batch_size}, max_length={args.max_length}"
)
# TODO: if using --runs=x txt2img_obj.log will output on each display every iteration infos from the start
# TODO: if using --batch_count=x txt2img_obj.log will output on each display every iteration infos from the start
text_output += txt2img_obj.log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"