fix checkpoint loading for electra, return attentions#157

sfluegel05 merged 1 commit intodevChEB-AI/python-chebai:devfrom

fix/load-electra-checkpointsChEB-AI/python-chebai:fix/load-electra-checkpointsCopy head branch name to clipboard