fix a bug in samping function

This commit is contained in:
marcoyang 2023-10-04 00:09:27 +08:00
parent e058ba0a65
commit 7c56d8f06b

View File

@ -370,7 +370,7 @@ def triplet_text_sampling_with_context_list(
# get the normalized text and pre_text # get the normalized text and pre_text
text = transforms[i_text](gt_text) text = transforms[i_text](gt_text)
pre_text = get_pre_text_with_context_list2( pre_text = get_pre_text_with_context_list2(
text=gt_pre_text, text=gt_text,
pre_text=gt_pre_text, pre_text=gt_pre_text,
context_list=context_list, context_list=context_list,
rare_words_list=rare_word_list, rare_words_list=rare_word_list,