load_ift_gpt_model

llms_from_scratch_rs::examples::ch07::addons

Function load_ift_gpt_model 

Source 
pub fn load_ift_gpt_model<P>(
    varmap: &mut VarMap,
    vb: VarBuilder<'_>,
    cfg: Config,
    checkpoint_path: &P,
) -> Result<GPTModel>
where
    P: AsRef<Path> + Debug,
Expand description

Helper function to load reference and policy models