V3nator
/

mar_test2

Unconditional Image Generation

Model card Files Files and versions Community

LimiTrix commited on Jan 22

Commit

95b91ff

·

1 Parent(s): 8f129d0

test

Files changed (1) hide show

mar.py +1 -1

mar.py CHANGED Viewed

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

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