[−][src]Struct rust_bert::pipelines::generation::OpenAIGenerator
Methods
impl OpenAIGenerator
[src]
pub fn new(generate_config: GenerateConfig) -> Fallible<OpenAIGenerator>
[src]
Build a new OpenAIGenerator
Arguments
generate_config
-GenerateConfig
object containing the resource references (model, vocabulary, configuration), generation options and device placement (CPU/GPU)
Example
use rust_bert::pipelines::generation::{GenerateConfig, OpenAIGenerator}; let generate_config = GenerateConfig { max_length: 30, do_sample: true, num_beams: 5, temperature: 1.1, num_return_sequences: 3, ..Default::default() }; let gpt_generator = OpenAIGenerator::new(generate_config)?;
Trait Implementations
impl LanguageGenerator<OpenAIGPTLMHeadModel, OpenAiGptVocab, OpenAiGptTokenizer> for OpenAIGenerator
[src]
Auto Trait Implementations
impl !RefUnwindSafe for OpenAIGenerator
impl !Send for OpenAIGenerator
impl !Sync for OpenAIGenerator
impl Unpin for OpenAIGenerator
impl !UnwindSafe for OpenAIGenerator
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,
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, 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.
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>,