Merge remote-tracking branch 'upstream/main'

MarcusLoppe · Jul 2, 2024 · c3f7c9f · c3f7c9f
2 parents b99b081 + 1b244c6
commit c3f7c9f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/meshgpt_pytorch/version.py b/meshgpt_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.5.5'
+__version__ = '1.5.7'
diff --git a/setup.py b/setup.py
@@ -20,6 +20,7 @@
     'mesh generation'
   ],
   install_requires=[
+    'matplotlib',
     'accelerate>=0.25.0',
     'huggingface_hub>=0.21.4',
     'beartype',
@@ -33,14 +34,13 @@
     'jaxtyping',
     'local-attention>=1.9.0',
     'numpy',
-    'matplotlib',
-    'pytorch-custom-utils>=0.0.20',
+    'pytorch-custom-utils>=0.0.9',
     'taylor-series-linear-attention>=0.1.6',
     'torch>=2.1',
     'torch_geometric',
     'tqdm',
     'vector-quantize-pytorch>=1.14.22',
-    'x-transformers>=1.30.19',
+    'x-transformers>=1.30.19,<1.31',
   ],
   setup_requires=[
     'pytest-runner',