Problem has been resolved from my end. This was the fix I had applied:
Copy paste the code from the below mentioned link to the NVIDIA’s version of pooling.py sentence-transformers/sentence_transformers/models/Pooling.py at master · UKPLab/sentence-transformers · GitHub