test
Browse files
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
|