pub struct RunConfig {
pub model_params_b: f32,
pub hidden_dim: u32,
pub num_layers: u32,
pub num_heads: u32,
pub batch_size: u32,
pub seq_len: u32,
pub cuda_graphs: bool,
pub quant_type: QuantType,
pub kernel_type: KernelType,
}Expand description
Runtime configuration for feature extraction
Fields§
§model_params_b: f32§num_layers: u32§num_heads: u32§batch_size: u32§seq_len: u32§cuda_graphs: bool§quant_type: QuantType§kernel_type: KernelTypeTrait Implementations§
Source§impl<'de> Deserialize<'de> for RunConfig
impl<'de> Deserialize<'de> for RunConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for RunConfig
impl RefUnwindSafe for RunConfig
impl Send for RunConfig
impl Sync for RunConfig
impl Unpin for RunConfig
impl UnwindSafe for RunConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more