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
|