search

design-edit / DesignEdit

Code for DesignEdit
MIT License
288 stars 21 forks source link

Differences between paper and huggingface demo #4

Open LL3RD opened 2 months ago

LL3RD commented 2 months ago

Thank you for your open source work！When I tried the remove task of huggingface demo, I found that it was not consistent with the picture reported in the paper. The pikachu gets one hand multi-inpainted when removing the apple. Do you have any idea?

image image

jiayueru commented 2 months ago

Thank you very much for your observation. In the Hugging Face demo, at line 243 of DesignEdit/src/demo/model.py, we set attend_scale to dilate the edges of the original mask. You can adjust the corresponding parameters during local deployment to achieve better results. We will also update the Hugging Face demo to make this parameter adjustable as soon as possible.

image