Enum rust_bert::pipelines::common::ConfigOption

pub enum ConfigOption {
Show 24 variants
    Bart(BartConfig),
    Bert(BertConfig),
    DistilBert(DistilBertConfig),
    Deberta(DebertaConfig),
    DebertaV2(DebertaV2Config),
    Electra(ElectraConfig),
    Marian(MarianConfig),
    MobileBert(MobileBertConfig),
    OpenAiGpt(OpenAiGptConfig),
    T5(T5Config),
    LongT5(LongT5Config),
    Albert(AlbertConfig),
    XLNet(XLNetConfig),
    GPT2(Gpt2Config),
    GPTJ(GptJConfig),
    Reformer(ReformerConfig),
    Roberta(RobertaConfig),
    ProphetNet(ProphetNetConfig),
    Longformer(LongformerConfig),
    Pegasus(PegasusConfig),
    GPTNeo(GptNeoConfig),
    MBart(MBartConfig),
    M2M100(M2M100Config),
    FNet(FNetConfig),
}

Expand description

Abstraction that holds a model configuration, can be of any of the supported models

Variants§

Bart(BartConfig)

Bart configuration

Bert(BertConfig)

Bert configuration

DistilBert(DistilBertConfig)

DistilBert configuration

Deberta(DebertaConfig)

DeBERTa configuration

DebertaV2(DebertaV2Config)

DeBERTa V2 configuration

Electra(ElectraConfig)

Electra configuration

Marian(MarianConfig)

Marian configuration

MobileBert(MobileBertConfig)

MobileBert configuration

OpenAiGpt(OpenAiGptConfig)

OpenAI GPT configuration

T5(T5Config)

T5 configuration

LongT5(LongT5Config)

LongT5 configuration

Albert(AlbertConfig)

Albert configuration

XLNet(XLNetConfig)

XLNet configuration

GPT2(Gpt2Config)

GPT2 configuration

GPTJ(GptJConfig)

GPT-J configuration

Reformer(ReformerConfig)

Reformer configuration

Roberta(RobertaConfig)

RoBERTa configuration

ProphetNet(ProphetNetConfig)

ProphetNet configuration

Longformer(LongformerConfig)

Longformer configuration

Pegasus(PegasusConfig)

Pegasus configuration

GPTNeo(GptNeoConfig)

GPT-Neo configuration

MBart(MBartConfig)

MBart configuration

M2M100(M2M100Config)

M2M100 configuration

FNet(FNetConfig)

FNet configuration

Implementations§

impl ConfigOption

pub fn from_file<P: AsRef<Path>>(model_type: ModelType, path: P) -> Self

Interface method to load a configuration from file

pub fn get_label_mapping(&self) -> &HashMap<i64, String>

pub fn get_max_len(&self) -> Option<i64>

pub fn get_vocab_size(&self) -> i64

pub fn get_decoder_start_token_id(&self) -> Option<i64>

pub fn get_forced_bos_token_id(&self) -> Option<i64>

pub fn get_forced_eos_token_id(&self) -> Option<i64>

Trait Implementations§

impl TryFrom<&ConfigOption> for AlbertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for BertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for DistilBertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for T5Config

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

Auto Trait Implementations§

impl RefUnwindSafe for ConfigOption

impl Send for ConfigOption

impl Sync for ConfigOption

impl Unpin for ConfigOption

impl UnwindSafe for ConfigOption

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Pointable for T

const ALIGN: usize = _

The alignment of pointer.

type Init = T

The type for initializers.

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more

impl<T> Same for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more