Function fastvit 

pub fn fastvit(
    cfg: &Config,
    nclasses: usize,
    vb: VarBuilder<'_>,
) -> Result<Func<'static>>