Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-01-10 08:25:00 -05:00
parent 663b0ed800
commit ac2adaa299
12 changed files with 2 additions and 24 deletions
-4
View File
@@ -681,10 +681,6 @@ class VectorQuantizer(nn.Module):
if self.unknown_index == "extra":
self.unknown_index = self.re_embed
self.re_embed = self.re_embed + 1
print(
f"Remapping {self.n_e} indices to {self.re_embed} indices. "
f"Using {self.unknown_index} for unknown indices."
)
else:
self.re_embed = n_e