lvwerra HF staff commited on
Commit
0fd0d1f
1 Parent(s): 5c34853

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
utils.py CHANGED
@@ -229,6 +229,8 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
229
  add_generation_prompt=True
230
  )
231
  model_input = tokenizer.decode(input_tokens)
 
 
232
 
233
  response_stream = client.text_generation(
234
  model=model,
 
229
  add_generation_prompt=True
230
  )
231
  model_input = tokenizer.decode(input_tokens)
232
+
233
+ print(f"Model input:\n{model_input}\n{'='*80}")
234
 
235
  response_stream = client.text_generation(
236
  model=model,