Skip to main content

SaveLoad

rust_ai_core::ecosystem::peft

Trait SaveLoad 

Source 
pub trait SaveLoad {
    // Required methods
    fn state_dict(&self) -> Result<HashMap<String, Tensor>, PeftError>;
    fn load_state_dict(
        &mut self,
        state_dict: HashMap<String, Tensor>,
    ) -> Result<(), PeftError>;
}
Expand description

Trait for adapters that can be saved and loaded.

Required Methods§

Source

fn state_dict(&self) -> Result<HashMap<String, Tensor>, PeftError>

Get all adapter tensors as a map of name -> tensor.

§Errors

Returns an error if tensor retrieval fails.

Source

fn load_state_dict( &mut self, state_dict: HashMap<String, Tensor>, ) -> Result<(), PeftError>

Load adapter tensors from a state dict.

§Errors

Returns an error if tensor loading fails.

Implementors§