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

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +1 -1
dc.py CHANGED
@@ -714,7 +714,6 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
714
  pass
715
  progress(1, desc="Model loaded.")
716
  progress(0, desc="Starting Inference...")
717
- images = None
718
  for info_state, stream_images, info_images in sd_gen_generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
719
  guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
720
  lora4, lora4_wt, lora5, lora5_wt, sampler,
@@ -729,6 +728,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
729
  ):
730
  images = stream_images
731
  progress(1, desc="Inference completed.")
 
732
  output_image = images[0][0] if images else None
733
 
734
  return output_image
 
714
  pass
715
  progress(1, desc="Model loaded.")
716
  progress(0, desc="Starting Inference...")
 
717
  for info_state, stream_images, info_images in sd_gen_generate_pipeline(prompt, negative_prompt, 1, num_inference_steps,
718
  guidance_scale, True, generator, lora1, lora1_wt, lora2, lora2_wt, lora3, lora3_wt,
719
  lora4, lora4_wt, lora5, lora5_wt, sampler,
 
728
  ):
729
  images = stream_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