Cache aware streaming infer error with size mismatched
**Describe the bug**

I trained a Cache aware streaming Conformer CTC model, and decode with
```
CUDA_VISIBLE_DEVICES="0" python examples/asr/asr_cache_aware_streaming/speech_to_text_cache_aware_streaming_infer.py \
    --asr_model=/home/xx/NeMo/exp/libri_mini_conformer_ctc_small_streaming/Conformer-CTC-BPE/2023-06-25_01-05-04/checkpoints/Conformer-CTC-BPE.nemo \
    --manifest_file=/home/xx/NeMo/data/libri_960h/dev_clean_2.json \
    --batch_size=1 \
    --use_amp \
    --debug_mode
```

but got a error:
```
  File "/home/xx/NeMo/nemo/collections/asr/parts/submodules/multi_head_attention.py", line 211, in forward
    key, value, query = self.update_cache(key=key, value=value, query=query, cache=cache, cache_next=cache_next)
  File "/home/xx/NeMo/nemo/collections/asr/parts/submodules/multi_head_attention.py", line 154, in update_cache
    cache_next[self._cache_id, :, -q_keep_size:, :] = query[:, :q_keep_size, :]
RuntimeError: The expanded size of the tensor (9988) must match the existing size (200) at non-singleton dimension 1.  Target sizes: [1, 9988, 176].  Tensor sizes: [200, 176]
```

**Steps/Code to reproduce bug**
Use lated NeMo code and use code like below to train a streaming model:
```
python examples/asr/asr_ctc/speech_to_text_ctc_bpe.py \
    --config-path /home/xx/NeMo/examples/asr/conf/conformer \
    --config-name conformer_ctc_bpe_small.yaml \
    model.train_ds.manifest_filepath=/home/xx/NeMo/data/libri_960h/train_clean_100
    model.validation_ds.manifest_filepath=/home/xx/NeMo/data/libri_960h/dev_clean.
    model.tokenizer.dir=/home/xx/NeMo/data/eng_bpe_128 \  # extracted from pretrained model stt_en_conformer_ctc_small_ls
    model.tokenizer.type=bpe \
    model.optim.lr=0.1 \
    model.optim.sched.warmup_steps=20000 \
    trainer.devices=2 \
    trainer.accelerator="gpu" \
    trainer.strategy="ddp" \
    trainer.max_epochs=100 \
    exp_manager.exp_dir=/home/xx/NeMo/exp/libri_mini_conformer_ctc_small_streaming
```

and then use code like below to decode:
```
CUDA_VISIBLE_DEVICES="0" python examples/asr/asr_cache_aware_streaming/speech_to_text_cache_aware_streaming_infer.py \
    --asr_model=/home/xx/NeMo/exp/libri_mini_conformer_ctc_small_streaming/Conformer-CTC-BPE/2023-06-25_01-05-04/checkpoints/Conformer-CTC-BPE.nemo \
    --manifest_file=/home/xx/NeMo/data/libri_960h/dev_clean_2.json \
    --batch_size=1 \
    --use_amp \
    --debug_mode
```

**Expected behavior**

Normal decoding.

**Environment details**

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:
- Red Hat 8.4.1-1
- PyTorch 1.11.0
- Python 3.8.17

**Additional context**
full log: 
[xx.log](https://github.com/NVIDIA/NeMo/files/11859985/xx.log)
