test

mar.py

@@ -272,8 +272,7 @@ class MARBert(nn.Module):
         # generate latents
         for step in indices:
             cur_tokens = tokens.clone()
-            print(cur_tokens.shape)
-
+            print("test")
             # class embedding and CFG
             if labels is not None:
                 class_embedding = self.class_emb(labels)