diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index 6e98474..31a44e4 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -56,7 +56,7 @@ def num_to_groups(num, divisor): arr.append(remainder) return arr -def convert_image_to(img_type, image): +def convert_image_to_fn(img_type, image): if image.mode != img_type: return image.convert(img_type) return image @@ -704,7 +704,7 @@ class Dataset(Dataset): self.image_size = image_size self.paths = [p for ext in exts for p in Path(f'{folder}').glob(f'**/*.{ext}')] - maybe_convert_fn = partial(convert_image_to, convert_image_to) if exists(convert_image_to) else nn.Identity() + maybe_convert_fn = partial(convert_image_to_fn, convert_image_to) if exists(convert_image_to) else nn.Identity() self.transform = T.Compose([ T.Lambda(maybe_convert_fn), diff --git a/setup.py b/setup.py index 0ecd2f6..6bc32d8 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ from setuptools import setup, find_packages setup( name = 'denoising-diffusion-pytorch', packages = find_packages(), - version = '0.27.10', + version = '0.27.11', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',