Enum rust_bert::pipelines::text_generation::TextGenerationOption [−][src]
Variants
GPT2(GPT2Generator)Text Generator based on GPT2 model
GPT(OpenAIGenerator)Text Generator based on GPT model
XLNet(XLNetGenerator)Text Generator based on XLNet model
Reformer(ReformerGenerator)Text Generator based on Reformer model
Implementations
impl TextGenerationOption[src]
pub fn new(config: TextGenerationConfig) -> Result<Self, RustBertError>[src]
pub fn model_type(&self) -> ModelType[src]
Returns the ModelType for this TextGenerationOption
pub fn get_tokenizer(&self) -> &TokenizerOption[src]
Interface method to access tokenizer
pub fn generate_indices<'a, S>(
&self,
prompt_texts: Option<S>,
attention_mask: Option<Tensor>,
min_length: Option<i64>,
max_length: Option<i64>
) -> Vec<Vec<i64>> where
S: AsRef<[&'a str]>, [src]
&self,
prompt_texts: Option<S>,
attention_mask: Option<Tensor>,
min_length: Option<i64>,
max_length: Option<i64>
) -> Vec<Vec<i64>> where
S: AsRef<[&'a str]>,
Interface method to generate() of the particular models.
Auto Trait Implementations
impl !RefUnwindSafe for TextGenerationOption
impl Send for TextGenerationOption
impl !Sync for TextGenerationOption
impl Unpin for TextGenerationOption
impl UnwindSafe for TextGenerationOption
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized, [src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized, [src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized, [src]
T: ?Sized,
pub fn borrow_mut(&mut self) -> &mut T[src]
impl<T> From<T> for 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]
U: From<T>,
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<U> for T where
U: Into<T>, [src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
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]
U: TryFrom<T>,
type Error = <U as TryFrom<T>>::Error
The type returned in the event of a conversion 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>,
V: MultiLane<T>,