return_dict_in_generate = new_generation_config.return_dict_in_generate if new_generation_config.do_sample: return self._beam_sample( input_ids, 24 changes: 5 additions & 19 deletions 24 intel_extension_for_pytorch/transformers/generation/greedy_search.py Original file line numberDiff line...