bash''' model = DiffusionModel() load_checkpoint = torch.load("Diffusion_model1-2.pth") model.load_checkpoint(load_checkpoint) '''