Skip to content
Snippets Groups Projects
Commit 2aff67fa authored by Gregor Kobsik's avatar Gregor Kobsik
Browse files

fix bug: remove debug output

parent 2e494f10
No related branches found
No related tags found
No related merge requests found
......@@ -31,8 +31,6 @@ class BasicGenerator():
"""
stop_idx = len(val)
print(f"start_idx: {start_idx}, stop_idx: {stop_idx}")
# sample tokens autoregressively
for token_idx in tqdm(
range(start_idx, stop_idx, self.num_tokens), initial=start_idx, total=stop_idx, leave=False, desc="Tokens"
......@@ -49,7 +47,6 @@ class BasicGenerator():
# check transformer token capacity
if len(logits) <= token_idx:
print("Reached max num of tokens (generator):", f"logits: {len(logits)}", f"token_idx: {token_idx}")
return val[:token_idx] # reached maximum number of tokens
# compute token probabilities from logits
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment