Add support for non-square output

castorini / daam

Diffusion attentive attribution maps for interpreting Stable Diffusion.

MIT License

689 stars 63 forks source link

Add support for non-square output #26

Open ArmastusChen opened 1 year ago

ArmastusChen commented 1 year ago

Hi,

Thanks for the great work! I wonder if you can add the support for non-square image outputs? For example, the output height is 704 and width is 512. That will be helpful. Thank you!

daemon commented 1 year ago

Sure thing.