Skip to main content

vit_encode

Function vit_encode 

Source
pub fn vit_encode(a: &VitArgs<'_>, x: &mut [f32]) -> bool
Expand description

The encoder layers of a vision tower resident on the device, x ([n, d]) in place. false = refused; the caller runs its per-op path. wgpu only for now; CMF_VIT_RESIDENT=0 declines.