KuangDW commited on
Commit
ba3dd45
·
1 Parent(s): eb490a1

session_id delete

Browse files
Files changed (1) hide show
  1. vecalign/plan2align.py +2 -2
vecalign/plan2align.py CHANGED
@@ -563,7 +563,7 @@ def generate_windows(source, translations):
563
  adjusted_mt_list.append(adjusted_mt.copy())
564
  mt_index += 1
565
 
566
- clear_folder(temp_folder, session_id)
567
  return adjusted_src, adjusted_mt_list
568
 
569
  ################################# main function #################################
@@ -618,7 +618,7 @@ def translate_text(text, session_id, model, tokenizer, device,
618
  else:
619
  translations = translate_with_deepinfra(model, tokenizer, device, source_sentence, buffer, good_ref_contexts_num + iteration, SRC_LANGUAGE, TASK_LANGUAGE)
620
 
621
- src_windows, mt_windows_list = generate_windows(source_sentence, translations, SESSION_ID)
622
  # print("Evaluate translations and update buffer ..............")
623
 
624
  src_context_list = list(src_windows)
 
563
  adjusted_mt_list.append(adjusted_mt.copy())
564
  mt_index += 1
565
 
566
+ clear_folder(temp_folder, SESSION_ID)
567
  return adjusted_src, adjusted_mt_list
568
 
569
  ################################# main function #################################
 
618
  else:
619
  translations = translate_with_deepinfra(model, tokenizer, device, source_sentence, buffer, good_ref_contexts_num + iteration, SRC_LANGUAGE, TASK_LANGUAGE)
620
 
621
+ src_windows, mt_windows_list = generate_windows(source_sentence, translations)
622
  # print("Evaluate translations and update buffer ..............")
623
 
624
  src_context_list = list(src_windows)