Skip to content
Snippets Groups Projects
Commit 46d5a373 authored by Renato Vukovic's avatar Renato Vukovic
Browse files

update inference script

parent 44ee6ff0
No related branches found
No related tags found
No related merge requests found
......@@ -154,7 +154,7 @@ def main():
else:
LLM_input = prompt_generator.generate_prompt(step=i, dialogue=text, term_list=terms, relations_so_far=relations_so_far, additional_input=current_responses)
try: #if it fails save the current dict and then throw the error
try: #if it fails save the current dict and then log the error and continue
relationlist=["has slot", "has value", "has domain", "refers to same concept as"]
if config.only_hasslot:
......@@ -179,7 +179,7 @@ def main():
logger.info(f"Checkpoint saved at {checkpoint_filename} after {counter} dialogues")
logger.error(f"Error at dialogue {dial_id} in split {split}")
logger.error(f"Error message: {e}")
e.with_traceback()
continue
if config.predict_for_cot_decoding:
output_string += "Step " + str(i) + " response:\n"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment