From d4965e556201d94bf500a9e7bf1fe065b6c20c92 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Tue, 5 Dec 2023 08:42:47 -0800
Subject: [PATCH] oops

---
 meshgpt_pytorch/meshgpt_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
index 606cb9d2..ae41a87f 100644
--- 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