Amr Keleg commited on
Commit
17680bc
·
1 Parent(s): 383fc79

Tiny black edit

Browse files
Files changed (1) hide show
  1. eval_utils.py +1 -1
eval_utils.py CHANGED
@@ -69,7 +69,7 @@ def prompt_chat_LLM(model, tokenizer, text):
69
  },
70
  ]
71
  input_ids = tokenizer.apply_chat_template(
72
- messages, tokenize=True, add_generation_prompt=True, return_tensors="pt"
73
  )
74
  gen_tokens = model.generate(input_ids, max_new_tokens=20)
75
  gen_text = tokenizer.decode(gen_tokens[0], skip_special_tokens=True)
 
69
  },
70
  ]
71
  input_ids = tokenizer.apply_chat_template(
72
+ messages, tokenize=True, add_generation_prompt=True, return_tensors="pt",
73
  )
74
  gen_tokens = model.generate(input_ids, max_new_tokens=20)
75
  gen_text = tokenizer.decode(gen_tokens[0], skip_special_tokens=True)