For model and code

[riolys]

Hi! Excellent work achieving sota on custom diffusion.
When would you like to release your code and pre-trained model?
I cant wait for testing it.