update kornia

lucidrains · Jun 13, 2021 · 169e069 · 169e069
1 parent 0e14694
commit 169e069
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
@@ -22,7 +22,7 @@
 from PIL import Image
 import torchvision
 from torchvision import transforms
-from kornia import filter2D
+from kornia import filter2d
 
 from lightweight_gan.diff_augment import DiffAugment
 from lightweight_gan.version import __version__
@@ -190,7 +190,7 @@ def __init__(self):
     def forward(self, x):
         f = self.f
         f = f[None, None, :] * f [None, :, None]
-        return filter2D(x, f, normalized=True)
+        return filter2d(x, f, normalized=True)
 
 # attention
 

diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.20.1'
+__version__ = '0.20.2'
diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
     'adabelief-pytorch',
     'einops>=0.3',
     'fire',
-    'kornia',
+    'kornia>=0.5.4',
     'numpy',
     'pillow',
     'retry',