fix property codebook

lucidrains · Nov 20, 2021 · f96d197 · f96d197
1 parent 826c620
commit f96d197
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vector_quantize_pytorch/vector_quantize_pytorch.py b/vector_quantize_pytorch/vector_quantize_pytorch.py
@@ -276,7 +276,7 @@ def __init__(
 
     @property
     def codebook(self):
-        return self._codebook.codebook
+        return self._codebook.embed
 
     def forward(self, x):
         need_transpose = not self.channel_last