Struct LoraAdapter 

pub struct LoraAdapter { /* private fields */ }

Manages LoRA adaptation for a collection of layers

Implementations

impl LoraAdapter

pub fn new(config: LoraConfig) -> Self

Create a new LoRA adapter with the given configuration

pub fn add_layer( &mut self, name: String, weight: Array2<f32>, ) -> ModelResult<()>

Add a layer to the adapter with the given name and weight matrix

pub fn forward_layer( &self, name: &str, input: &Array1<f32>, ) -> ModelResult<Array1<f32>>

Forward pass through a named layer

pub fn merge_all(&mut self) -> ModelResult<()>

Merge all LoRA weights into the original weight matrices

pub fn unmerge_all(&mut self) -> ModelResult<()>

Unmerge all LoRA weights from the original weight matrices

pub fn total_trainable_params(&self) -> usize

Total trainable parameters across all layers

pub fn total_original_params(&self) -> usize

Total original (frozen) parameters across all layers

pub fn overall_compression_ratio(&self) -> f32

Overall compression ratio

pub fn layer_names(&self) -> Vec<&str>

Get names of all layers

pub fn get_layer(&self, name: &str) -> Option<&LoraLinear>

Get an immutable reference to a named layer

pub fn get_layer_mut(&mut self, name: &str) -> Option<&mut LoraLinear>

Get a mutable reference to a named layer

pub fn config(&self) -> &LoraConfig

Get the adapter configuration

pub fn summary(&self) -> LoraAdapterSummary

Get a summary of the adapter

Trait Implementations

impl Clone for LoraAdapter

fn clone(&self) -> LoraAdapter

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LoraAdapter

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

Formats the value using the given formatter.