Enum tokenizers::processors::template::Piece

pub enum Piece {
    Sequence {
        id: Sequence,
        type_id: u32,
    },
    SpecialToken {
        id: String,
        type_id: u32,
    },
}

Represents the different kind of pieces that constitute a template. It can be either the input sequence or a SpecialToken:

• The Sequence has an associated type_id which is used by default for any token inside this sequence. The Sequence corresponds to one of the input sequence given as input of the PostProcessor.

• The SpecialToken has an associated id. It corresponds to a SpecialToken.

The easiest way to build a Piece is actually by converting it from a string:

let sequence_with_type_id_0 = Piece::try_from("$0").unwrap();
let sequence_with_type_id_1 = Piece::try_from("$1").unwrap();
let special_token_cls = Piece::try_from("[CLS]").unwrap();

Variants

Sequence

Fields

id: Sequence

type_id: u32

SpecialToken

Fields

id: String

type_id: u32

Trait Implementations

impl Clone for Piece

fn clone(&self) -> Piece

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Piece

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Piece

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for Piece

fn eq(&self, other: &Piece) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for Piece

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl TryFrom<&str> for Piece

type Error = String

The type returned in the event of a conversion error.

fn try_from(s: &str) -> StdResult<Self, Self::Error>

Performs the conversion.

impl TryFrom<String> for Piece

type Error = String

The type returned in the event of a conversion error.

fn try_from(s: String) -> StdResult<Self, Self::Error>

Performs the conversion.

impl Eq for Piece

impl StructuralPartialEq for Piece