Spaces:
Runtime error
Runtime error
Update semanticist/stage2/generate.py
Browse files
semanticist/stage2/generate.py
CHANGED
@@ -26,6 +26,7 @@ def decode_n_tokens(
|
|
26 |
input_pos += 1
|
27 |
new_tokens.append(next_token.clone())
|
28 |
cur_token = next_token
|
|
|
29 |
|
30 |
return new_tokens
|
31 |
|
|
|
26 |
input_pos += 1
|
27 |
new_tokens.append(next_token.clone())
|
28 |
cur_token = next_token
|
29 |
+
print(f'generating {i} of {num_new_tokens} new tokens')
|
30 |
|
31 |
return new_tokens
|
32 |
|