ControlNet for Stable Diffusion 2.1

[jdp8]

I tried to use the conv_sd_to_onnx.py script to convert thibaud's Canny ControlNet for SD 2.1 to ONNX format but it did not work, specifically a shape mismatch error occurred. The command I used was python conv_sd_to_onnx.py --model_path "stabilityai/stable-diffusion-2-1" --controlnet_path "thibaud/controlnet-sd21-canny-diffusers" --output_path "models/sd21-canny" --attention-slicing auto --ckpt-upcast-attention --fp16.

It would be great if the conversion of ControlNet for SD 2.1 to ONNX format was possible. For now, I am using ControlNet 1.0 and 1.1 with SD 1.5 in ONNX and it's enough but it would be great to have ControlNet for SD 2.1.