Upload dc.py
Browse files
dc.py
CHANGED
|
@@ -537,11 +537,11 @@ class GuiSD:
|
|
| 537 |
)
|
| 538 |
if save_generated_images:
|
| 539 |
info_images += f"<br>{download_links}"
|
| 540 |
-
|
| 541 |
## BEGIN MOD
|
|
|
|
| 542 |
img = save_images(img, metadata)
|
|
|
|
| 543 |
## END MOD
|
| 544 |
-
|
| 545 |
info_state = "COMPLETE"
|
| 546 |
|
| 547 |
yield info_state, img, info_images
|
|
|
|
| 537 |
)
|
| 538 |
if save_generated_images:
|
| 539 |
info_images += f"<br>{download_links}"
|
|
|
|
| 540 |
## BEGIN MOD
|
| 541 |
+
if not isinstance(img, list): img = [img]
|
| 542 |
img = save_images(img, metadata)
|
| 543 |
+
img = [(i, None) for i in img]
|
| 544 |
## END MOD
|
|
|
|
| 545 |
info_state = "COMPLETE"
|
| 546 |
|
| 547 |
yield info_state, img, info_images
|