PhysiQuanty commited on
Commit
09956ff
·
verified ·
1 Parent(s): 6504615

Update inference.py

Browse files
Files changed (1) hide show
  1. 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) # debug (tu supprimeras avant commit)
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)