Add support for pix2pix and inpainting models
Opened this issue · 2 comments
ljleb commented
Merging a pix2pix or inpainting model with a normal model yet does not work. We need to catch dimension mismatches in the shapes of the tensors and take appropriate action. We need to look around on github, I'm pretty sure there is code for this already somewhere.
Enferlain commented
ljleb commented
Yes, but I'm not sure this is how it's supposed to be done. Instead of ignoring the keys, we should test slicing, merging and concat. I forgot to specify this in the OP, thanks.