Enum ModelInputs 

pub enum ModelInputs {
    Text {
        input_ids: Tensor,
        attention_mask: Option<Tensor>,
        position_ids: Option<Tensor>,
    },
    Image {
        image_tensor: Tensor,
        image_features: Option<Tensor>,
    },
    Multimodal {
        input_ids: Tensor,
        attention_mask: Option<Tensor>,
        image_tensor: Option<Tensor>,
        token_type_ids: Option<Tensor>,
    },
    Audio {
        input_features: Tensor,
        decoder_input_ids: Option<Tensor>,
    },
}

Model inputs for forward pass

Variants

Text

Fields

input_ids: Tensor

attention_mask: Option<Tensor>

position_ids: Option<Tensor>

Image

Fields

image_tensor: Tensor

image_features: Option<Tensor>

Multimodal

Fields

input_ids: Tensor

attention_mask: Option<Tensor>

image_tensor: Option<Tensor>

token_type_ids: Option<Tensor>

Audio

Fields

input_features: Tensor

decoder_input_ids: Option<Tensor>

Implementations

impl ModelInputs

pub fn text(text: &str) -> Self

pub fn image(image_tensor: Tensor) -> Self

Trait Implementations

impl Clone for ModelInputs

fn clone(&self) -> ModelInputs

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelInputs

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

Formats the value using the given formatter.