From 3b9f0f757d6dab96a2ed650946fc778c68d58b77 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Mon, 23 Oct 2023 08:34:06 -0700 Subject: [PATCH] just follow the 80/20% rule in software design --- denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 2 +- denoising_diffusion_pytorch/version.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index e0b6aee5c..81ef4ad14 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -873,7 +873,7 @@ def __init__( amp = False, mixed_precision_type = 'fp16', split_batches = True, - convert_image_to = None, + convert_image_to = 'RGB', calculate_fid = True, inception_block_idx = 2048, max_grad_norm = 1., diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py index 4175d3998..be0b61075 100644 --- a/denoising_diffusion_pytorch/version.py +++ b/denoising_diffusion_pytorch/version.py @@ -1 +1 @@ -__version__ = '1.9.2' +__version__ = '1.9.3'