Upload dc.py
Browse files
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
|