mlx_rs::module

Trait ModuleParametersExt

Source
pub trait ModuleParametersExt: ModuleParameters {
    // Provided methods
    fn eval(&self) -> Result<(), Exception> { ... }
    fn load_safetensors(
        &mut self,
        path: impl AsRef<Path>,
    ) -> Result<(), IoError> { ... }
    fn save_safetensors(&self, path: impl AsRef<Path>) -> Result<(), IoError> { ... }
}
Expand description

Extension trait for ModuleParameters. This is implemented for all types that implement ModuleParameters.

Provided Methods§

Source

fn eval(&self) -> Result<(), Exception>

Evaluate the module parameters.

Source

fn load_safetensors(&mut self, path: impl AsRef<Path>) -> Result<(), IoError>

Load module parameters from a safetensors file.

Source

fn save_safetensors(&self, path: impl AsRef<Path>) -> Result<(), IoError>

Save module parameters to a file in safetensors format.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§