search

Fantasy-Studio / Paint-by-Example

Paint by Example: Exemplar-based Image Editing with Diffusion Models
https://arxiv.org/abs/2211.13227
Other
1.06k stars 96 forks source link

Encountered an error #10

Closed deepdreamerzhang closed 1 year ago

deepdreamerzhang commented 1 year ago

Hello, very interesting work! When I ran inference.py, I got Paint-by Example/ldm/models/diffusion/ddim.py", line 150, in ddim_sampling img = torch.cat((img, kwargs['rest']), dim=1) KeyError: 'rest'. Do you have any suggestions about this, thank you!

Fantasy-Studio commented 1 year ago

Please use "--plms" for sampling. I will update the "ddim.py" in the future.

Fantasy-Studio commented 1 year ago

The bug in ddim.py has been solved, you can now also try to use ddim for sampling.