pub fn tiny_vit_5m(vb: VarBuilder<'_>) -> Result<TinyViT>