Method get_layers

python/TIDE/adapters/base.py:14–16

Return the list of transformer decoder layers.

(self, model: nn.Module)

    @abstractmethod
    def get_layers(self, model: nn.Module) -> list[nn.Module]:
        """Return the list of transformer decoder layers."""
        ...

    @abstractmethod
    def get_hidden_state(self, layer_output) -> torch.Tensor: