Incorporating nuclear quantum effects for investigating non-adiabatic dynamics