Struct Model 

pub struct Model {
    pub embed_prompts: Embedding,
    pub enc_to_dec_proj: Option<Linear>,
    pub decoder: Decoder,
    pub text_encoder: T5EncoderModel,
    pub decoder_start_token_id: u32,
    pub pad_token_id: u32,
    pub audio_encoder: Model,
}

Fields

embed_prompts: Embedding

enc_to_dec_proj: Option<Linear>

decoder: Decoder

text_encoder: T5EncoderModel

decoder_start_token_id: u32

pad_token_id: u32

audio_encoder: Model

Implementations

impl Model

pub fn new(cfg: &Config, vb: VarBuilder<'_>) -> Result<Self>

pub fn generate( &mut self, prompt_tokens: &Tensor, description_tokens: &Tensor, lp: LogitsProcessor, max_steps: usize, ) -> Result<Tensor>

Note that the returned tensor uses the CPU device.

Trait Implementations

impl Clone for Model

fn clone(&self) -> Model

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Model

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.