Enum rust_bert::pipelines::summarization::SummarizationOption[−][src]

pub enum SummarizationOption {
    Bart(BartGenerator),
    T5(T5Generator),
    ProphetNet(ProphetNetConditionalGenerator),
}

Abstraction that holds one particular summarization model, for any of the supported models

Variants

Bart(BartGenerator)

Summarizer based on BART model

T5(T5Generator)

Summarizer based on T5 model

ProphetNet(ProphetNetConditionalGenerator)

Summarizer based on ProphetNet model

Implementations

`impl SummarizationOption`[src]

`pub fn new(config: SummarizationConfig) -> Result<Self, RustBertError>`[src]

`pub fn model_type(&self) -> ModelType`[src]

Returns the ModelType for this SummarizationOption

`pub fn generate<'a, S>( &self, prompt_texts: Option<S>, attention_mask: Option<Tensor> ) -> Vec<String> where S: AsRef<[&'a str]>,` [src]

Interface method to generate() of the particular models.

Auto Trait Implementations

`impl !RefUnwindSafe for SummarizationOption`

`impl Send for SummarizationOption`

`impl !Sync for SummarizationOption`

`impl Unpin for SummarizationOption`

`impl UnwindSafe for SummarizationOption`

Blanket Implementations

`impl<T> Any for T where T: 'static + ?Sized,` [src]

`pub fn type_id(&self) -> TypeId`[src]

`impl<T> Borrow<T> for T where T: ?Sized,` [src]

`pub fn borrow(&self) -> &T`[src]

`impl<T> BorrowMut<T> for T where T: ?Sized,` [src]

`pub fn borrow_mut(&mut self) -> &mut T`[src]

`impl<T> From<T> for T`[src]

`pub fn from(t: T) -> T`[src]

`impl<T> Instrument for T`[src]

`pub fn instrument(self, span: Span) -> Instrumented<Self>`[src]

`pub fn in_current_span(self) -> Instrumented<Self>`[src]

`impl<T, U> Into for T where U: From<T>,` [src]

`pub fn into(self) -> U`[src]

`impl<T> Pointable for T`

`pub const ALIGN: usize`

`type Init = T`

The type for initializers.

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

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

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

`pub unsafe fn drop(ptr: usize)`

`impl<T> Same<T> for T`

`type Output = T`

Should always be Self

`impl<T, U> TryFrom for T where U: Into<T>,` [src]

`type Error = Infallible`

The type returned in the event of a conversion error.

`pub fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>`[src]

`impl<T, U> TryInto for T where U: TryFrom<T>,` [src]

`type Error = >::Error`

The type returned in the event of a conversion error.

Enum rust_bert::pipelines::summarization::SummarizationOption[−][src]

Abstraction that holds one particular summarization model, for any of the supported models

Variants

Implementations

impl SummarizationOption[src]

pub fn new(config: SummarizationConfig) -> Result<Self, RustBertError>[src]

pub fn model_type(&self) -> ModelType[src]

pub fn generate<'a, S>( &self, prompt_texts: Option<S>, attention_mask: Option<Tensor>) -> Vec<String> where S: AsRef<[&'a str]>, [src]

Auto Trait Implementations

impl !RefUnwindSafe for SummarizationOption

impl Send for SummarizationOption

impl !Sync for SummarizationOption

impl Unpin for SummarizationOption

impl UnwindSafe for SummarizationOption

Blanket Implementations

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

pub fn type_id(&self) -> TypeId[src]

impl<T> Borrow<T> for T where T: ?Sized, [src]

pub fn borrow(&self) -> &T[src]

impl<T> BorrowMut<T> for T where T: ?Sized, [src]

pub fn borrow_mut(&mut self) -> &mut T[src]

impl<T> From<T> for T[src]

pub fn from(t: T) -> T[src]

impl<T> Instrument for T[src]

pub fn instrument(self, span: Span) -> Instrumented<Self>[src]

pub fn in_current_span(self) -> Instrumented<Self>[src]

impl<T, U> Into<U> for T where U: From<T>, [src]

pub fn into(self) -> U[src]

impl<T> Pointable for T

pub const ALIGN: usize

type Init = T

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

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

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

pub unsafe fn drop(ptr: usize)

impl<T> Same<T> for T

type Output = T

impl<T, U> TryFrom<U> for T where U: Into<T>, [src]

type Error = Infallible

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

impl<T, U> TryInto<U> for T where U: TryFrom<T>, [src]

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

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

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

pub fn vzip(self) -> V