Amr Keleg
commited on
Commit
·
17680bc
1
Parent(s):
383fc79
Tiny black edit
Browse files- 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)
|