How to test erasing concept of NSFW finetuned models?

[hjinnkim]

Thank you for your interesting papers. I'm really impressed by your research.

I want to test the generation result of NSFW-nudity prompts with the provided finetuned model weights. But, when I sample the images with the prompts 'nudity', the model generate the nude images!

How can I generate images which 'nudity' is erased?

[rohitgandikota]

Thank you! I am not sure about your inference setup. Could you try and use our generation code provided in the eval-scripts folder? https://github.com/rohitgandikota/erasing/blob/main/eval-scripts/generate-images.py

If this still persists, please do let us know

[rohitgandikota]

closing this issue because of inactivity. Feel free to open it if the issue still persists