From 27988aaf5b0ae571f98cade17bd779ed6d6166a3 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Fri, 11 Oct 2024 07:15:37 -0700 Subject: [PATCH] turn on a breakthrough technique for vq --- meshgpt_pytorch/meshgpt_pytorch.py | 1 + meshgpt_pytorch/version.py | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py index ab4e330..7488e95 100644 --- 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 index ef2e1d5..bcd8d54 100644 --- 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 index d6e7a72..6b1a87a 100644 --- 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=[