trustformers_models::weight_loading::config

Struct WeightLoadingConfig

pub struct WeightLoadingConfig {
    pub format: Option<WeightFormat>,
    pub lazy_loading: bool,
    pub memory_mapped: bool,
    pub streaming: bool,
    pub device: String,
    pub dtype: WeightDataType,
    pub quantization: Option<QuantizationConfig>,
    pub cache_dir: Option<PathBuf>,
    pub verify_checksums: bool,
    pub distributed: Option<DistributedConfig>,
}

Expand description

Weight loading configuration

Fields§

§format: Option<WeightFormat>§lazy_loading: bool§memory_mapped: bool§streaming: bool§device: String§dtype: WeightDataType§quantization: Option<QuantizationConfig>§cache_dir: Option<PathBuf>§verify_checksums: bool§distributed: Option<DistributedConfig>

WeightLoadingConfig

Struct WeightLoadingConfig Copy item path

Fields§

Trait Implementations§

impl Clone for WeightLoadingConfig

fn clone(&self) -> WeightLoadingConfig

fn clone_from(&mut self, source: &Self)

impl Debug for WeightLoadingConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

impl Default for WeightLoadingConfig

fn default() -> Self

Auto Trait Implementations§

impl Freeze for WeightLoadingConfig

impl RefUnwindSafe for WeightLoadingConfig

impl Send for WeightLoadingConfig

impl Sync for WeightLoadingConfig

impl Unpin for WeightLoadingConfig

impl UnsafeUnpin for WeightLoadingConfig

impl UnwindSafe for WeightLoadingConfig

Blanket Implementations§

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for Twhere T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for Twhere T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> CloneToUninit for Twhere T: Clone,

unsafe fn clone_to_uninit(&self, dest: *mut u8)

impl<T> From<T> for T

fn from(t: T) -> T

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

fn in_current_span(self) -> Instrumented<Self>

impl<T, U> Into<U> for Twhere U: From<T>,

fn into(self) -> U

impl<T> IntoEither for T

fn into_either(self, into_left: bool) -> Either<Self, Self>

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>where F: FnOnce(&Self) -> bool,

impl<T> Pointable for T

const ALIGN: usize

type Init = T

unsafe fn init(init: <T as Pointable>::Init) -> usize

unsafe fn deref<'a>(ptr: usize) -> &'a T

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

unsafe fn drop(ptr: usize)

impl<T> PolicyExt for Twhere T: ?Sized,

fn and<P, B, E>(self, other: P) -> And<T, P>where T: Policy<B, E>, P: Policy<B, E>,

fn or<P, B, E>(self, other: P) -> Or<T, P>where T: Policy<B, E>, P: Policy<B, E>,

impl<T> Same for T

type Output = T

impl<T> ToOwned for Twhere T: Clone,

type Owned = T

fn to_owned(&self) -> T

fn clone_into(&self, target: &mut T)

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for Twhere V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>where S: Into<Dispatch>,

fn with_current_subscriber(self) -> WithDispatch<Self>

Struct WeightLoadingConfig

impl<T> Any for T
where T: 'static + ?Sized,

impl<T> Borrow<T> for T
where T: ?Sized,

impl<T> BorrowMut<T> for T
where T: ?Sized,

impl<T> CloneToUninit for T
where T: Clone,

impl<T, U> Into<U> for T
where U: From<T>,

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

impl<T> PolicyExt for T
where T: ?Sized,

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Policy<B, E>, P: Policy<B, E>,

impl<T> ToOwned for T
where T: Clone,

impl<T, U> TryFrom<U> for T
where U: Into<T>,

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,