Struct candle_lora::LoraEmbedding

pub struct LoraEmbedding { /* private fields */ }

Implementations

impl LoraEmbedding

pub fn new( old: &dyn EmbeddingLayerLike, embed_config: &LoraEmbeddingConfig, config: &LoraConfig, vb: &VarBuilder<'_>, id: usize, ) -> Result<Self>

Trait Implementations

impl Clone for LoraEmbedding

fn clone(&self) -> LoraEmbedding

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for LoraEmbedding

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

Formats the value using the given formatter.

impl EmbeddingLayerLike for LoraEmbedding

fn embeddings(&self) -> &Tensor

fn hidden_size(&self) -> usize

impl Merge for LoraEmbedding

fn get_delta_weight(&self) -> Result<Tensor, MergeErrorOrError>

Get the delta weight of the LoRA layer. This is meant to be an internal method.

fn merge_weights(&mut self) -> Result<(), MergeErrorOrError>

Merge the LoRA weights.

fn unmerge_weights(&mut self) -> Result<(), MergeErrorOrError>

Unmerge the LoRA weights.

impl Module for LoraEmbedding

fn forward(&self, input: &Tensor) -> Result<Tensor>

impl Saveable for LoraEmbedding

fn get_tensors(&self, accum: &mut HashMap<String, Tensor>)