Facing an issue while infer a prompt-tuned model "TypeError: megatron_gpt_generate() takes 5 positional arguments but 6 were given" 
**Describe the bug**

While we try to infer/use a prompt-tuned model we are getting an error "TypeError: megatron_gpt_generate() takes 5 positional arguments but 6 were given" 

**Steps/Code to reproduce bug**
We have followed below notebook for prompt-tuning:
https://github.com/NVIDIA/NeMo/blob/main/tutorials/nlp/Multitask_Prompt_and_PTuning.ipynb

While we tried to use prompt-tuned model using following lines of code: 
model.generate(inputs=test_examples, length_params=None)

**Expected behavior**

It should give the result of given prompt.

**Environment overview (please complete the following information)**

 - Environment location: Google Colab Pro with GPU and High RAM
 - Method of NeMo install: [from source]. !python -m pip install git+https://github.com/NVIDIA/NeMo.git@{BRANCH}#egg=nemo_toolkit[all]
 
**Environment details**

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:
- OS version: Ubuntu 18.04.6 LTS
- PyTorch version: 1.12.1+cu113
- Python version: Python 3.7.14


