There are three head models for the former use case:

[LukeForEntityClassification], for tasks to classify a single entity in an input text such as
    entity typing, e.g. the Open Entity dataset.
    This model places a linear head on top of the output entity representation.