From e2492578c71ddfbc6440379183bc3c104dd4e60e Mon Sep 17 00:00:00 2001 From: JJJYmmm Date: Wed, 10 Apr 2024 19:40:55 +0800 Subject: [PATCH] [fix]: Fix variable naming errors --- opensora/models/ae/imagebase/vqvae/quantize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opensora/models/ae/imagebase/vqvae/quantize.py b/opensora/models/ae/imagebase/vqvae/quantize.py index dd258efe9..148062e9e 100644 --- a/opensora/models/ae/imagebase/vqvae/quantize.py +++ b/opensora/models/ae/imagebase/vqvae/quantize.py @@ -365,8 +365,8 @@ class EMAVectorQuantizer(nn.Module): def __init__(self, n_embed, embedding_dim, beta, decay=0.99, eps=1e-5, remap=None, unknown_index="random"): super().__init__() - self.codebook_dim = codebook_dim - self.num_tokens = num_tokens + self.codebook_dim = embedding_dim + self.num_tokens = n_embed self.beta = beta self.embedding = EmbeddingEMA(self.num_tokens, self.codebook_dim, decay, eps)