enforce latest torch

lucidrains · Feb 3, 2022 · 0262a9d · 0262a9d
1 parent e8f8700
commit 0262a9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.20.7'
+__version__ = '0.20.8'
diff --git a/setup.py b/setup.py
@@ -31,7 +31,7 @@
     'numpy',
     'pillow',
     'retry',
-    'torch>=1.6',
+    'torch>=1.10',
     'torchvision',
     'tqdm'
   ],