fix a bug thanks to @lonzi

ZeroCool940711 · Jan 16, 2023 · 3f9de5c · 3f9de5c
1 parent f7ea957
commit 3f9de5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/muse_maskgit_pytorch/muse_maskgit_pytorch.py b/muse_maskgit_pytorch/muse_maskgit_pytorch.py
@@ -510,7 +510,7 @@ def forward(
         # get loss
 
         ce_loss = self.transformer(
-            ids,
+            x,
             texts = texts,
             text_embeds = text_embeds,
             conditioning_token_ids = cond_token_ids,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'muse-maskgit-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.18',
+  version = '0.0.19',
   license='MIT',
   description = 'MUSE - Text-to-Image Generation via Masked Generative Transformers, in Pytorch',
   author = 'Phil Wang',