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

[SD] Fix few things in sendTo feature (#1132)

Browse Source
This commit is contained in:
Gaurav Shukla
2023-03-02 22:41:55 +05:30
committed by GitHub
parent 911dff16f1
commit b118f183d1
10 changed files with 46 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/stable_diffusion/scripts/img2img.py
View File

@@ -220,7 +220,7 @@ def img2img_inf(
text_output += img2img_obj.log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"
return generated_imgs, generated_imgs[0], text_output
return generated_imgs, text_output
if __name__ == "__main__":

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

@@ -175,7 +175,7 @@ def inpaint_inf(
text_output += inpaint_obj.log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"
return generated_imgs, generated_imgs[0], text_output
return generated_imgs, text_output
if __name__ == "__main__":

2
apps/stable_diffusion/scripts/outpaint.py
View File

@@ -190,7 +190,7 @@ def outpaint_inf(
text_output += outpaint_obj.log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"
return generated_imgs, generated_imgs[0], text_output
return generated_imgs, text_output
if __name__ == "__main__":

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

@@ -172,7 +172,7 @@ def txt2img_inf(
# text_output += txt2img_obj.log
text_output += f"\nTotal image generation time: {total_time:.4f}sec"
yield generated_imgs, generated_imgs[0], text_output
yield generated_imgs, text_output
if __name__ == "__main__":