Resole merge conflicts

This commit is contained in:
Fred de Gier 2022-10-03 14:27:32 +02:00
commit de71bb6ff5
6 changed files with 53 additions and 35 deletions

View file

@ -111,7 +111,7 @@ class CodeGenProxy:
if stop_words is None:
stop_words = []
if stop_words:
stop_word_list = np.repeat(to_word_list_format([stop_words], self.tokenizer), input_start_ids.shape[0],
stop_word_list = np.repeat(self.to_word_list_format([stop_words], self.tokenizer), input_start_ids.shape[0],
axis=0)
else:
stop_word_list = np.concatenate([np.zeros([input_start_ids.shape[0], 1, 1]).astype(