Update README.md
Browse files
README.md
CHANGED
@@ -65,27 +65,6 @@ def inference_control_redux(style_image, control_image, save_image, size=(1024,
|
|
65 |
dst.paste(image, (size[0] * 2, 0))
|
66 |
dst.save(save_image)
|
67 |
|
68 |
-
def inference_control(style_image, control_image, save_image, size=(1024, 1024)):
|
69 |
-
style_image = load_image(style_image).resize(size)
|
70 |
-
control_image = load_image(control_image).resize(size)
|
71 |
-
pipe_prior_redux_out = pipe_prior_redux(style_image, return_dict=False)
|
72 |
-
prompt_embeds, pooled_prompt_embeds = pipe_prior_redux_out[0], pipe_prior_redux_out[1]
|
73 |
-
image = pipe(
|
74 |
-
control_image=control_image,
|
75 |
-
height=size[1],
|
76 |
-
width=size[0],
|
77 |
-
num_inference_steps=50,
|
78 |
-
guidance_scale=30.0,
|
79 |
-
prompt_embeds=prompt_embeds,
|
80 |
-
pooled_prompt_embeds=pooled_prompt_embeds,
|
81 |
-
).images[0]
|
82 |
-
|
83 |
-
dst = Image.new('RGB', (size[0] * 3, size[1]))
|
84 |
-
dst.paste(style_image, (0, 0))
|
85 |
-
dst.paste(control_image, (size[0], 0))
|
86 |
-
dst.paste(image, (size[0] * 2, 0))
|
87 |
-
dst.save(save_image)
|
88 |
-
|
89 |
style_image = "style.png" # download from this repo
|
90 |
control_image = "control.png"
|
91 |
save_image = "save.png"
|
|
|
65 |
dst.paste(image, (size[0] * 2, 0))
|
66 |
dst.save(save_image)
|
67 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
68 |
style_image = "style.png" # download from this repo
|
69 |
control_image = "control.png"
|
70 |
save_image = "save.png"
|