unconditional image generation part 1 : Generate images from pipeline

generate.py

def generate(pretrained_pipe_dir):

    pipeline = DDPMPipeline.from_pretrained(pretrained_pipe_dir).to("cuda")
    images = pipeline(
        batch_size=1,
        generator=torch.manual_seed(123),
    ).images

    return images 
  
  generate("path/to/saved/pipeline")