V3nator
/

mar_test2

Unconditional Image Generation

Model card Files Files and versions Community

LimiTrix commited on Jan 22

Commit

75b8440

·

1 Parent(s): 95b91ff

test

Files changed (1) hide show

mar.py +1 -1

mar.py CHANGED Viewed

@@ -272,7 +272,7 @@ class MAR(nn.Module):
         # generate latents
         for step in indices:
             cur_tokens = tokens.clone()
-            print(cur_tokens)
             # class embedding and CFG
             if labels is not None:

         # generate latents
         for step in indices:
             cur_tokens = tokens.clone()
+            print(cur_tokens.shape)
             # class embedding and CFG
             if labels is not None: