Struct PeftModel 

pub struct PeftModel<A>
where
    A: Adapter,
{ /* private fields */ }

PEFT model wrapper for managing adapters across modules.

Provides module-level adapter management with pattern-based targeting.

Implementations

impl<A> PeftModel<A>

where A: Adapter,

pub fn new() -> PeftModel<A>

Create a new PEFT model wrapper.

pub fn add_adapter<F>( &mut self, adapter_name: impl Into<String>, pattern: &str, module_names: &[&str], adapter_factory: F, ) -> Result<usize, PeftError>

where F: Fn(&str) -> Result<A, PeftError>,

Add an adapter to modules matching the given pattern.

Arguments

• adapter_name - Unique name for the adapter
• pattern - Pattern to match module names
• module_names - List of all module names in the model
• adapter_factory - Function to create adapter instances

Errors

Returns an error if adapter creation fails

pub fn set_adapter( &mut self, module_name: &str, adapter_name: &str, ) -> Result<(), PeftError>

Set the active adapter for a specific module.

Errors

Returns an error if the module or adapter doesn’t exist

pub fn set_adapter_all( &mut self, adapter_name: impl Into<String>, ) -> Result<(), PeftError>

Set the active adapter for all modules.

Errors

Returns an error if the adapter doesn’t exist in any module

pub fn active_adapter_name(&self) -> Option<&str>

Get the active adapter name.

pub fn adapter_names(&self) -> &[String]

Get all registered adapter names.

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

Get module names that have adapters.

pub fn has_adapter(&self, module_name: &str) -> bool

Check if a module has any adapters.

pub fn forward_module( &self, module_name: &str, input: &Tensor, base_output: Option<&Tensor>, ) -> Result<Tensor, PeftError>

Forward pass for a specific module.

Arguments

• module_name - Name of the module
• input - Input tensor
• base_output - Optional base layer output

Errors

Returns an error if module not found or no active adapter

pub fn get_adapter( &self, module_name: &str, adapter_name: &str, ) -> Result<&A, PeftError>

Get a reference to an adapter for a module.

Errors

Returns an error if module or adapter not found

pub fn num_parameters(&self) -> usize

Get the total number of trainable parameters across all active adapters.

pub fn num_modules(&self) -> usize

Get the number of modules with adapters.

Trait Implementations

impl<A> Default for PeftModel<A>

where A: Adapter,

fn default() -> PeftModel<A>

Returns the “default value” for a type.