diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py index 58d478a..3f262a6 100644 --- a/lightweight_gan/version.py +++ b/lightweight_gan/version.py @@ -1 +1 @@ -__version__ = '1.2.0' +__version__ = '1.2.1' diff --git a/setup.py b/setup.py index 8ec6475..1d7cab6 100644 --- a/setup.py +++ b/setup.py @@ -25,13 +25,13 @@ ], install_requires=[ 'adabelief-pytorch', - 'einops>=0.3', + 'einops>=0.8.0', 'fire', 'kornia>=0.5.4', 'numpy', 'pillow', 'retry', - 'torch>=1.10', + 'torch>=2.2', 'torchvision', 'tqdm' ],