From c52689d6ab23a79a647074abb7ea13ff98a0940d Mon Sep 17 00:00:00 2001 From: lucidrains Date: Thu, 8 Feb 2024 14:44:53 -0800 Subject: [PATCH] latest ema-pytorch for the post-hoc ema proposed by Karras et al. --- denoising_diffusion_pytorch/version.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py index 8047ba3fd..192e7e0f9 100644 --- a/denoising_diffusion_pytorch/version.py +++ b/denoising_diffusion_pytorch/version.py @@ -1 +1 @@ -__version__ = '1.10.7' +__version__ = '1.10.8' diff --git a/setup.py b/setup.py index caffbd900..ec301a79f 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ install_requires=[ 'accelerate', 'einops', - 'ema-pytorch', + 'ema-pytorch>=0.4.2', 'numpy', 'pillow', 'pytorch-fid',