Would it make sense to create a class LatentDiffusion(torch.nn.Module) and then use composition i.e. create a wrapper class ComposerLatentDiffusion(ComposerModel) that holds a reference to a LatentDiffusion and delegates forward etc. to it?
Reasoning: Try to cleanly separate Composer related parts from PRX pipeline logic.
Originally posted by @photoroman in #7 (comment)
Would it make sense to create a class
LatentDiffusion(torch.nn.Module)and then use composition i.e. create a wrapper classComposerLatentDiffusion(ComposerModel)that holds a reference to aLatentDiffusionand delegatesforwardetc. to it?Reasoning: Try to cleanly separate Composer related parts from PRX pipeline logic.
Originally posted by @photoroman in #7 (comment)