turn on a breakthrough technique for vq

lucidrains · Oct 11, 2024 · 27988aa · 27988aa
1 parent 208519f
commit 27988aa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/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

diff --git a/meshgpt_pytorch/version.py b/meshgpt_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.5.16'
+__version__ = '1.6.0'
diff --git a/setup.py b/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=[