turn on a breakthrough technique for vq

Author: lucidrains

meshgpt_pytorch/meshgpt_pytorch.py

@@ -590,6 +590,7 @@ def __init__(
                 codebook_size = codebook_size,
                 shared_codebook = True,
                 commitment_weight = 1.,
+                rotation_trick = True,
                 stochastic_sample_codes = rvq_stochastic_sample_codes,
                 **rvq_kwargs,
                 **rq_kwargs

meshgpt_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.5.16'
+__version__ = '1.6.0'

setup.py

@@ -39,7 +39,7 @@
     'torch>=2.1',
     'torch_geometric',
     'tqdm',
-    'vector-quantize-pytorch>=1.14.22',
+    'vector-quantize-pytorch>=1.18.1',
     'x-transformers>=1.30.19,<1.31',
   ],
   setup_requires=[