From 7e1455e42f6ad19d5bed33c0c1a6f3d0af243bff Mon Sep 17 00:00:00 2001 From: Ben Evans Date: Mon, 6 Feb 2023 18:19:39 +0000 Subject: [PATCH] Run same number of DDPM steps in inference as training Resolves ValueError: `num_inference_steps`: 1000 cannot be larger than `self.config.train_timesteps`: 50 as the unet model trained with this scheduler can only handle maximal 50 timesteps. --- .../unconditional_image_generation/train_unconditional.py | 1 + examples/unconditional_image_generation/train_unconditional.py | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py b/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py index b9dec9a745e8..586ed8c8d397 100644 --- a/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py +++ b/examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py @@ -552,6 +552,7 @@ def transform_images(examples): generator=generator, batch_size=args.eval_batch_size, output_type="numpy", + num_inference_steps=args.ddpm_num_steps, ).images # denormalize the images and save to tensorboard diff --git a/examples/unconditional_image_generation/train_unconditional.py b/examples/unconditional_image_generation/train_unconditional.py index 324ddb0538e3..32dd96c88249 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -587,6 +587,7 @@ def transform_images(examples): generator=generator, batch_size=args.eval_batch_size, output_type="numpy", + num_inference_steps=args.ddpm_num_steps, ).images # denormalize the images and save to tensorboard