force einops 0.6.1 or greater and call allow_ops_in_compiled_graph

2025-12-19 09:44:19 +01:00 · 2023-04-20 14:08:52 -07:00
parent 0069857cf8
commit 00e07b7d61
3 changed files with 9 additions and 2 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',
+    'einops>=0.6.1',
    'embedding-reader',
    'kornia>=0.5.4',
    'numpy',