Method encode

opencompass/models/base.py:111–122

Encode prompt to tokens. Not necessary for most cases. Args: prompt (str): Input string. Returns: torch.Tensor: Encoded tokens.

(self, prompt: str)

    @abstractmethod
    def encode(self, prompt: str) -> torch.Tensor:
        """Encode prompt to tokens. Not necessary for most cases.

        Args:
            prompt (str): Input string.

        Returns:
            torch.Tensor: Encoded tokens.
        """
        raise NotImplementedError(
            f'{self.__class__.__name__} does not implement'
            '`encode` method.')

    @abstractmethod
    def decode(self, tokens: torch.Tensor) -> str: