rewriting a deep generative model