cene555/Imagen-pytorch

any plans for the training codes?

Opened this issue · 5 comments

any plans for the training codes?

I've had limited success modifying the code based off of this commit, but caveat emptor. I ended up deciding to wait for a future update, so don't have much additional context to add.

d0e3e4b

main purpose is to train from scratch, does not rely on interference lol.

Training is possible under the commit referenced in my previous comment, if you're willing to modify the code to fit your specific environment. Inference code is included in the current (latest) commit.

@lupinetine Did you get good training results?