Spaces:
Runtime error
Runtime error
修复img2img的报错
#4
by
HaloMaster
- opened
app.py
CHANGED
@@ -25,7 +25,7 @@ pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(dev
|
|
25 |
def infer_text2img(prompt, guide, steps, width, height, image_in, strength):
|
26 |
if image_in is not None:
|
27 |
init_image = image_in.convert("RGB").resize((width, height))
|
28 |
-
output = pipe_img2img(prompt,
|
29 |
else:
|
30 |
output = pipe_text2img(prompt, width=width, height=height, guidance_scale=guide, num_inference_steps=steps,)
|
31 |
image = output.images[0]
|
|
|
25 |
def infer_text2img(prompt, guide, steps, width, height, image_in, strength):
|
26 |
if image_in is not None:
|
27 |
init_image = image_in.convert("RGB").resize((width, height))
|
28 |
+
output = pipe_img2img(prompt, image=init_image, strength=strength, guidance_scale=guide, num_inference_steps=steps)
|
29 |
else:
|
30 |
output = pipe_text2img(prompt, width=width, height=height, guidance_scale=guide, num_inference_steps=steps,)
|
31 |
image = output.images[0]
|