LimiTrix commited on
Commit
8f129d0
·
1 Parent(s): f445293
Files changed (1) hide show
  1. mar.py +1 -0
mar.py CHANGED
@@ -271,6 +271,7 @@ class MAR(nn.Module):
271
  indices = tqdm(indices)
272
  # generate latents
273
  for step in indices:
 
274
  cur_tokens = tokens.clone()
275
 
276
  # class embedding and CFG
 
271
  indices = tqdm(indices)
272
  # generate latents
273
  for step in indices:
274
+ print(cur_tokens)
275
  cur_tokens = tokens.clone()
276
 
277
  # class embedding and CFG