Method get_lm_head

python/TIDE/adapters/base.py:29–31

Return the language model head (linear projection to vocab).

(self, model: nn.Module)

    @abstractmethod
    def get_lm_head(self, model: nn.Module) -> nn.Module:
        """Return the language model head (linear projection to vocab)."""
        ...

    @abstractmethod
    def get_router_input_dim(self, model: nn.Module) -> int: