Where is the diffusion model defined?

[Abraham190137]

Hello,
Thanks for the great implimination! It's a lot easier to follow than the paper's GitHub. I was looking through the code, and I can't find where the model used in the Diffusion Policy Class is defined, ie. the class which is called in the line:
model_output, self_latents = self.model(noised_img, times, return_latents = True)
Where would I find the model definition? Thanks!

[lucidrains]

@Abraham190137 this isn't done yet, but will be in the coming year

[Abraham190137]

Looking forward to it! Thanks for the response.