The pre-trained uncertainty quantification heads for the repo: https://github.com/IINemo/llm-uncertainty-head