GPT-2 example from README.md not working. (BS=1) about transformers-neuronx HOT 6 CLOSED

(aws_neuron_venv_pytorch) ubuntu@ip-10-0-10-149:~$ python test.py
.
Compiler status PASS
.
Compiler status PASS
Downloading (…)olve/main/vocab.json: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.04M/1.04M [00:00<00:00, 64.1MB/s]
Downloading (…)olve/main/merges.txt: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 456k/456k [00:00<00:00, 38.3MB/s]
Downloading (…)/main/tokenizer.json: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.36M/1.36M [00:00<00:00, 34.0MB/s]
Setting `pad_token_id` to `eos_token_id`:50256 for open-end generation.
['Hello, I\'m a language model, you know. So I\'ll give you an example.\n\nNow this is really important, and you know: if you\'re a programmer, you\'re going to have to understand these things. I don\'t mean programming, or programming about languages. I mean writing. And so the first thing I need to do is write a program for the program, and then I need to write a function. And that\'s a lot of stuff, and it\'s not something that you could just throw away.\n\nThen you don\'t have to write a compiler. You can write a language, and you can write a function, and they\'ll look at your code and say, "Okay, this is going to be a function."\n\nNow, because languages are so powerful, and because they\'re so complicated, if it\'s a function, then you can write a function, and you\'re going to be able to write a function, and you\'re going to be able to write a function. And so that\'s a big problem, and it\'s one that\'s really hard to fix.\n\nSo the first thing I need to do is write a program. And then I need to write a function, and I need to write']

(aws_neuron_venv_pytorch) ubuntu@ip-10-0-10-149:~$ pip list | grep neuron
aws-neuronx-runtime-discovery 2.9
libneuronxla                  0.5.391
neuronx-cc                    2.8.0.25+a3ad0f342
neuronx-distributed           0.1.0
neuronx-hwm                   2.8.0.3+2b7c6da39
torch-neuronx                 1.13.1.1.9.0
torch-xla                     1.13.1+torchneuron8
transformers-neuronx          0.5.58