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 |
|