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

[SD] Add sendTo Upscaler

Browse Source 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2023-03-16 19:09:29 +05:30
parent 7ffe20b1c2
commit 0225434389
8 changed files with 55 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -97,6 +97,8 @@ def upscaler_inf(
height,
width,
device,
use_lora=None,
use_stencil=None,
)
if (
not global_obj.get_sd_obj()
@@ -177,7 +179,7 @@ def upscaler_inf(
text_output += f"\nscheduler={args.scheduler}, device={device}"
text_output += f"\nsteps={steps}, noise_level={noise_level}, guidance_scale={guidance_scale}, seed={seeds}"
text_output += f"\nsize={height}x{width}, batch_count={batch_count}, batch_size={batch_size}, max_length={args.max_length}"
text_output += upscaler_obj.log
text_output += global_obj.get_sd_obj().log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"
yield generated_imgs, text_output