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'