remove einops exts for better pytorch 2.0 compile compatibility

2025-12-19 09:44:19 +01:00 · 2023-04-20 07:05:29 -07:00
parent 580274be79
commit 0069857cf8
4 changed files with 39 additions and 23 deletions
--- a/setup.py
+++ b/setup.py
@@ -30,8 +30,7 @@ setup(
    'clip-anytorch>=2.5.2',
    'coca-pytorch>=0.0.5',
    'ema-pytorch>=0.0.7',
-    'einops>=0.4',
-    'einops-exts>=0.0.3',
+    'einops>=0.6',
    'embedding-reader',
    'kornia>=0.5.4',
    'numpy',