RuntimeError: shape '[1, 1, 2, 2]' is invalid for input of size 5929

[Frosty-Jackal]

I tried to implement sd_v2.1+depth_map demo(in README.md) today!

After I downloaded "v2-1_512-ema-pruned.ckpt" , "dpt_hybrid-midas-501f0c75.pt" and "sd21_encD_depth_14m.ckpt" correctly, I modified 'sd21_encD_depth_14m.yaml'.

Then I copied code in "REAMDE.md" and tried to run, but here came maybe two problems:

1. [404 keys are missing from the model (hint processing and cross connections included)]
2. RuntimeError: shape '[1, 1, 2, 2]' is invalid for input of size 5929
   (from "File "ControlNet-XS/ldm/modules/encoders/modules.py", line 189, in text_transformer_forward x = r(x, attn_mask=attn_mask)" )

Help me!