search

rohitgandikota / unified-concept-editing

Unified Concept Editing in Diffusion Models
https://unified.baulab.info
MIT License
97 stars 3 forks source link

Pre-trained weights #1

Open mnpham0417 opened 1 year ago

mnpham0417 commented 1 year ago

Hi, thank you for the great work! May I ask if you are planning to release the pre-trained weights for artistic concepts, objects concepts, and NSFW concept?

rohitgandikota commented 1 year ago

Yes, we can release the weights soon. Also, it is super fast to train the models to erase from scratch.

Shinyzenith commented 5 months ago

Hi @rohitgandikota Would you mind releasing the pretrained weights? We would love to compare our research against your work.

rohitgandikota commented 5 months ago

Hi, we have fixed the repository dependancies and the training efficiency. You should be able to train the models in 1-3 minutes now. Please give it a try!