DeferHook

Type Alias DeferHook 

Source
pub type DeferHook = Arc<dyn Fn(&str, WeightDtype) -> bool + Send + Sync>;
Expand description

A model loader’s answer to “which tensors will I read from disk myself?”.

(tensor name, store dtype) -> defer. Arc because both checkpoint loaders hold it for the length of a load and the fast loader’s pre-flight and retain loop both call it; Send + Sync because the fast loader’s pipeline is threaded.

Aliased Type§

pub struct DeferHook { /* private fields */ }