John6666 commited on
Commit
eb7bff2
·
verified ·
1 Parent(s): e7f4dbb

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +2 -2
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