Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -183,7 +183,7 @@ def main() -> None:
|
|
| 183 |
ids = text_to_base2_digits(text) # 0/1 bits (MSB->LSB)
|
| 184 |
ids = wrap_base2_sequence_2(ids, args.bos, args.eos) # [BOS] bits [EOS]
|
| 185 |
ids = ids + [int(args.bos)] # reset latent: ...[EOS][BOS]
|
| 186 |
-
print("[+] IDS = ", ids)
|
| 187 |
return ids
|
| 188 |
|
| 189 |
prompt_ids = encode_prompt(args.prompt)
|
|
|
|
| 183 |
ids = text_to_base2_digits(text) # 0/1 bits (MSB->LSB)
|
| 184 |
ids = wrap_base2_sequence_2(ids, args.bos, args.eos) # [BOS] bits [EOS]
|
| 185 |
ids = ids + [int(args.bos)] # reset latent: ...[EOS][BOS]
|
| 186 |
+
print("[+] PROMPT IDS = ", ids)
|
| 187 |
return ids
|
| 188 |
|
| 189 |
prompt_ids = encode_prompt(args.prompt)
|