File size: 972 Bytes
40588a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from PIL import Image
import numpy as np
from diffusers import DEISMultistepScheduler
from pipeline_onnx_stable_diffusion_controlnet import OnnxStableDiffusionControlNetPipeline
import onnxruntime as ort

pose_image = Image.open(r"dance_pose.png")

opts = ort.SessionOptions()
opts.enable_cpu_mem_arena = False
opts.enable_mem_pattern = False

pipe = OnnxStableDiffusionControlNetPipeline.from_pretrained(
    "model/anyv3-fp16-autoslicing-cn_openpose",
    sess_options=opts, 
    provider="DmlExecutionProvider",
)

pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
prompt = "1girl, blonde, long dress, dancing, best quality"
seed=25
generator = np.random.RandomState(seed)

images = pipe(    
    prompt,
    pose_image,
    width=512,
    height=512,
    negative_prompt="monochrome, lowres, bad anatomy, worst quality, low quality",
    num_inference_steps=30,
    generator=generator,
).images[0]
images.save("controlnet-openpose-test.png")