[−][src]Enum rust_bert::pipelines::conversation::ConversationOption
Variants
GPT2(GPT2Generator)Conversation based on GPT2 model
Implementations
impl ConversationOption[src]
pub fn new(config: ConversationConfig) -> Result<Self, RustBertError>[src]
pub fn get_eos_id(&self) -> Result<i64, RustBertError>[src]
pub fn get_tokenizer(&self) -> &TokenizerOption[src]
pub fn model_type(&self) -> ModelType[src]
Returns the ModelType for this ConversationOption
pub fn generate_from_ids_and_past(
&self,
input_ids: Tensor,
attention_mask: Option<Tensor>
) -> Vec<Vec<i64>>[src]
&self,
input_ids: Tensor,
attention_mask: Option<Tensor>
) -> Vec<Vec<i64>>
Interface method to generate_from_ids_and_past() of the particular models.
Auto Trait Implementations
impl !RefUnwindSafe for ConversationOption[src]
impl Send for ConversationOption[src]
impl !Sync for ConversationOption[src]
impl Unpin for ConversationOption[src]
impl UnwindSafe for ConversationOption[src]
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, 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>,