Trait candle_lora::EmbeddingLayerLike

pub trait EmbeddingLayerLike: Module + Debug + Saveable + Send + Sync {
    // Required methods
    fn embeddings(&self) -> &Tensor;
    fn hidden_size(&self) -> usize;
}

Any layer that is embedding-like.

Required Methods

fn embeddings(&self) -> &Tensor

fn hidden_size(&self) -> usize

Implementations on Foreign Types

impl EmbeddingLayerLike for Embedding

fn embeddings(&self) -> &Tensor

fn hidden_size(&self) -> usize

Implementors

impl EmbeddingLayerLike for LoraEmbedding