fix can eos logic

lucidrains · Dec 5, 2023 · fa17e84 · fa17e84
1 parent 19e7109
commit fa17e84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -515,7 +515,7 @@ def generate(
         curr_length = codes.shape[-1]
 
         for i in range(curr_length, self.max_seq_len):
-            can_eos = (i + 1) % (self.num_quantizers * 3)  # only allow for eos to be decoded at the end of each face, defined as 3 vertices with D residusl VQ codes
+            can_eos = divisible_by(i + 1, self.num_quantizers * 3)  # only allow for eos to be decoded at the end of each face, defined as 3 vertices with D residusl VQ codes
 
             logits = self.forward(codes, return_loss = False)
             logits = logits[:, -1]

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'meshgpt-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.7',
+  version = '0.0.8',
   license='MIT',
   description = 'MeshGPT Pytorch',
   author = 'Phil Wang',