load_model

tiny_recursive_rs::models::loader

Function load_model 

Source 
pub fn load_model<P: AsRef<Path>>(
    config: TRMConfig,
    weights_path: P,
    device: &Device,
) -> Result<TinyRecursiveModel>
Expand description

Load model from safetensors file

§Arguments

  • config - Model configuration
  • weights_path - Path to safetensors file
  • device - Device to load model on

§Returns

Loaded TinyRecursiveModel