Why `.float()` is in `conformer_modules.py`?
https://github.com/NVIDIA/NeMo/blob/eae1684f7f33c2a18de9ecfa42ec7db93d39e631/nemo/collections/asr/parts/submodules/conformer_modules.py#L259

Hi! I'm wondering why `.float()` is here. It seems like it will break code if both model parameters and input in float16