oops

lucidrains · Dec 5, 2023 · d4965e5 · d4965e5
1 parent 7eb5d39
commit d4965e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -449,7 +449,7 @@ def __init__(
         self.num_quantizers = autoencoder.num_quantizers
 
         self.sos_token = nn.Parameter(torch.randn(dim))
-        self.eos_token_id = self.codebook_size + 1
+        self.eos_token_id = self.codebook_size
 
         # they use axial positional embeddings