John6666 commited on
Commit
88ab107
·
verified ·
1 Parent(s): f0ec643

Upload 3 files

Browse files
Files changed (2) hide show
  1. dc.py +0 -1
  2. modutils.py +1 -0
dc.py CHANGED
@@ -728,7 +728,6 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
728
  ):
729
  images = stream_images if isinstance(stream_images, list) else images
730
  progress(1, desc="Inference completed.")
731
- print(images)#
732
  output_image = images[0][0] if images else None
733
 
734
  return output_image
 
728
  ):
729
  images = stream_images if isinstance(stream_images, list) else images
730
  progress(1, desc="Inference completed.")
 
731
  output_image = images[0][0] if images else None
732
 
733
  return output_image
modutils.py CHANGED
@@ -330,6 +330,7 @@ def save_gallery_history(images, files, history_gallery, history_files, progress
330
 
331
 
332
  def save_image_history(image, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
 
333
  basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}"
334
  if image is None or (not isinstance(image, str) and not isinstance(image, Image.Image) and not isinstance(image, tuple)): return gr.update(), gr.update()
335
  if not gallery: gallery = []
 
330
 
331
 
332
  def save_image_history(image, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
333
+ print(image)
334
  basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}"
335
  if image is None or (not isinstance(image, str) and not isinstance(image, Image.Image) and not isinstance(image, tuple)): return gr.update(), gr.update()
336
  if not gallery: gallery = []