-
Notifications
You must be signed in to change notification settings - Fork 4
/
sd15_controlnet_canny.py
45 lines (39 loc) · 1.58 KB
/
sd15_controlnet_canny.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from sd_task.task_runner import run_inference_task
from sd_task.task_args import InferenceTaskArgs
from diffusers.utils import make_image_grid
from reference_image import get_controlnet_ref_image_dataurl
if __name__ == '__main__':
ref_image = get_controlnet_ref_image_dataurl()
prompt = ("best quality, ultra high res, photorealistic++++, 1girl, off-shoulder sweate, smiling, "
"faded ash gray messy bun hair+, border light, depth of field, looking at "
"viewer, closeup")
negative_prompt = ("paintings, sketches, worst quality+++++, low quality+++++, normal quality+++++, lowres, "
"normal quality, monochrome++, grayscale++, skin spots, acnes, skin blemishes, "
"age spot, glans")
args = {
"version": "2.0.0",
"base_model": {
"name": "runwayml/stable-diffusion-v1-5"
},
"prompt": prompt,
"negative_prompt": negative_prompt,
"task_config": {
"num_images": 9,
"safety_checker": False,
"steps": 40
},
"controlnet": {
"model": "lllyasviel/sd-controlnet-canny",
"image_dataurl": ref_image,
"preprocess": {
"method": "canny",
"args": {
"low_threshold": 50,
"high_threshold": 100
}
}
}
}
images = run_inference_task(InferenceTaskArgs.model_validate(args))
image_grid = make_image_grid(images, 3, 3)
image_grid.save("./data/sd15_controlnet_canny.png")