address issue with generation with 1 quantizer, thanks to @MarcusLoppe

lucidrains · lucidrains · commit 08427251a829 · 2024-06-01T09:32:06.000-07:00
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -1308,7 +1308,7 @@ def generate(
         # remove a potential extra token from eos, if breaked early
 
         code_len = codes.shape[-1]
-        round_down_code_len = code_len // self.num_quantizers * self.num_quantizers
+        round_down_code_len = code_len // self.num_vertices_per_face * self.num_vertices_per_face
         codes = codes[:, :round_down_code_len]
 
         # early return of raw residual quantizer codes
diff --git a/meshgpt_pytorch/version.py b/meshgpt_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.2.12'
+__version__ = '1.2.14'
