from diffusion_pytorch.diffusion_gaussian import GaussianDiffusion, Trainer
# from diffusion_pytorch.new_twobranch_model import Model as TwoBranchNewModel