Skip to main content

AdaLoraLayer

Struct AdaLoraLayer 

Source
pub struct AdaLoraLayer { /* private fields */ }
Expand description

AdaLoRA layer using SVD-based parameterization.

Uses W = W0 + P * Λ * Q where:

  • P: Left singular vectors (out_features × r)
  • Λ: Diagonal singular values (r)
  • Q: Right singular vectors (r × in_features)

This allows for dynamic rank allocation by zeroing out singular values.

Implementations§

Source§

impl AdaLoraLayer

Source

pub fn new( in_features: usize, out_features: usize, config: AdaLoraConfig, device: &Device, ) -> Result<AdaLoraLayer, PeftError>

Create a new AdaLoRA layer.

§Arguments
  • in_features - Input dimension
  • out_features - Output dimension
  • config - AdaLoRA configuration
  • device - Device to create tensors on
§Errors

Returns error if configuration is invalid or tensor initialization fails.

Source

pub fn current_rank(&self) -> usize

Get the current active rank.

Source

pub fn target_rank(&self) -> usize

Get the target rank.

Source

pub fn init_rank(&self) -> usize

Get the initial rank.

Source

pub fn scaling(&self) -> f64

Get the scaling factor.

Source

pub fn update_rank_mask( &mut self, importance_scores: &Tensor, budget: usize, ) -> Result<(), PeftError>

Update the rank mask based on importance scores.

§Arguments
  • importance_scores - Importance score for each rank [init_r]
  • budget - Number of ranks to keep
§Errors

Returns error if tensor operations fail.

Source

pub fn orthogonal_regularization(&self) -> Result<Tensor, PeftError>

Compute the orthogonal regularization loss.

Encourages P^T P ≈ I and Q Q^T ≈ I.

§Errors

Returns error if tensor operations fail.

Source

pub fn get_importance_scores(&self) -> Result<Tensor, PeftError>

Get the importance scores for rank allocation.

The importance is based on the magnitude of singular values.

§Errors

Returns error if tensor operations fail.

Trait Implementations§

Source§

impl Adapter for AdaLoraLayer

Source§

type Config = AdaLoraConfig

The configuration type for this adapter.
Source§

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

Forward pass applying the adapter transformation. Read more
Source§

fn num_parameters(&self) -> usize

Get the number of trainable parameters.
Source§

fn config(&self) -> &<AdaLoraLayer as Adapter>::Config

Get the adapter’s configuration.
Source§

impl Mergeable for AdaLoraLayer

Source§

fn merge(&self, base_weight: &Tensor) -> Result<Tensor, PeftError>

Merge adapter weights into base model weights. Read more
Source§

fn unmerge(&self, merged_weight: &Tensor) -> Result<Tensor, PeftError>

Unmerge adapter weights from merged weights. Read more
Source§

impl Trainable for AdaLoraLayer

Source§

fn register_parameters( &self, _var_map: &mut VarMap, _prefix: &str, ) -> Result<(), PeftError>

Register trainable parameters with the variable map. Read more
Source§

fn freeze(&mut self)

Freeze all adapter parameters (disable gradients).
Source§

fn unfreeze(&mut self)

Unfreeze all adapter parameters (enable gradients).
Source§

fn is_frozen(&self) -> bool

Check if the adapter is frozen.

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

impl<T> ErasedDestructor for T
where T: 'static,