Enum ModelDType

pub enum ModelDType {
    Auto,
    BF16,
    F16,
    F32,
}

DType for the model.

If the model is quantized, this is ignored so it is reasonable to use the Default impl.

Note: When using Auto, fallback pattern is: BF16 -> F16 -> 32

Variants

Auto

BF16

F16

F32

Trait Implementations

impl Clone for ModelDType

fn clone(&self) -> ModelDType

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ModelDType

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

Formats the value using the given formatter.

impl Default for ModelDType

fn default() -> ModelDType

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ModelDType

fn deserialize<__D>( __deserializer: __D, ) -> Result<ModelDType, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for ModelDType

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

Formats the value using the given formatter.

impl FromStr for ModelDType

type Err = String

The associated error which can be returned from parsing.

fn from_str(s: &str) -> Result<ModelDType, <ModelDType as FromStr>::Err>

Parses a string s to return a value of this type.

impl IntoPy<Py<PyAny>> for ModelDType

fn into_py(self, py: Python<'_>) -> Py<PyAny>

Performs the conversion.

fn type_output() -> TypeInfo

Extracts the type hint information for this type when it appears as a return value.

impl PartialEq for ModelDType

fn eq(&self, other: &ModelDType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl PyClass for ModelDType

type Frozen = False

Whether the pyclass is frozen.

impl PyTypeInfo for ModelDType

const NAME: &'static str = "ModelDType"

Class name.

const MODULE: Option<&'static str> = ::core::option::Option::None

Module name, if any.

fn type_object_raw(py: Python<'_>) -> *mut PyTypeObject

Returns the PyTypeObject instance for this type.

fn type_object_bound(py: Python<'_>) -> Bound<'_, PyType>

Returns the safe abstraction over the type object.

fn is_type_of_bound(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type or a subclass of this type.

fn is_exact_type_of_bound(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type.

impl TryIntoDType for ModelDType

fn try_into_dtype(&self, devices: &[&Device]) -> Result<DType, Error>

impl Copy for ModelDType

impl StructuralPartialEq for ModelDType