Update inductor.py

inductor.py

@@ -129,15 +129,12 @@ class BartInductor(object):
     def extract_words_for_tA_bart(self, tA, k=6, print_it = False):
         spans = [t.lower().strip() for t in tA[:-1].split('<mask>')]
         generated_ids = self.tokenizer([tA], padding='longest', return_tensors='pt')['input_ids'].to(device)
-        generated_ret = self.orion_instance_generator.generate(generated_ids, num_beams=max(120, k),
-                                            #num_beam_groups=max(120, k),
-                                            max_length=generated_ids.size(1) + 15,
-                                            num_return_sequences=max(120, k), #min_length=generated_ids.size(1),
-                                            #diversity_penalty=2.0,
-                                            #length_penalty= 0.8,
-                                            #early_stopping=True, bad_words_ids=bad_words_ids, no_repeat_ngram_size=2,
-                                            output_scores=True,
-                                            return_dict_in_generate=True)
+        generated_ret = self.orion_instance_generator.generate(generated_ids.astype(np.float32), num_beams=max(120, k),
+                                                       min_length=min_length,
+                                                       max_length=max_length,
+                                                       early_stopping=early_stopping,
+                                                       **generator_kwargs)
+
         summary_ids = generated_ret['sequences']
         probs = F.softmax(generated_ret['sequences_scores'])
         txts = [self.tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True) for g in summary_ids]