28e6f98
1
2
3
from diffusion_pytorch.diffusion_gaussian import GaussianDiffusion, Trainer # from diffusion_pytorch.new_twobranch_model import Model as TwoBranchNewModel