new einops is torch compile friendly

2025-12-19 09:44:19 +01:00 · 2023-10-18 15:45:09 -07:00
parent c6c3882dc1
commit 410a6144e1
3 changed files with 2 additions and 9 deletions
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@ setup(
    'clip-anytorch>=2.5.2',
    'coca-pytorch>=0.0.5',
    'ema-pytorch>=0.0.7',
-    'einops>=0.6.1',
+    'einops>=0.7.0',
    'embedding-reader',
    'kornia>=0.5.4',
    'numpy',