import torch import pytorch_lightning as pl class LatentDiffusionModel(pl.LightningModule): # TODO pass