neuralworm commited on
Commit
3f5ced9
Β·
1 Parent(s): ec0e835

initial commit

Browse files
Files changed (1) hide show
  1. gen.py +1 -0
gen.py CHANGED
@@ -182,6 +182,7 @@ def generate(event):
182
 
183
 
184
  output_text = tokenizer.decode(tokens[0], skip_special_tokens=False)
 
185
  user_prompt_length = len(f"<bos><start_of_turn>user\n{prompt}\n{event}<end_of_turn>\n<start_of_turn>model\n") # Calculate user prompt length
186
 
187
  json_start_index = output_text.find("<json>")
 
182
 
183
 
184
  output_text = tokenizer.decode(tokens[0], skip_special_tokens=False)
185
+ print(output_text)
186
  user_prompt_length = len(f"<bos><start_of_turn>user\n{prompt}\n{event}<end_of_turn>\n<start_of_turn>model\n") # Calculate user prompt length
187
 
188
  json_start_index = output_text.find("<json>")