Struct ModelSpec 

pub struct ModelSpec {
    pub name: String,
    pub task: ModelTask,
    pub source: ModelSource,
    pub files: Vec<ModelFileRequest>,
    pub metadata: BTreeMap<String, String>,
    pub repo_id: String,
    pub revision: String,
}

Generic model spec.

Fields

name: String

Human-readable name for this value.

task: ModelTask

The task value.

source: ModelSource

The model source value.

files: Vec<ModelFileRequest>

The files value.

metadata: BTreeMap<String, String>

Caller-defined model metadata.

repo_id: String

👎Deprecated:

use ModelSpec::source or ModelSpec::repo_id() instead

Deprecated compatibility field for Hugging Face model identifiers.

revision: String

👎Deprecated:

use ModelSpec::source or ModelSpec::revision() instead

Deprecated compatibility field for Hugging Face revisions.

Implementations

impl ModelSpec

pub fn new(repo_id: impl Into<String>, task: ModelTask) -> Self

Creates a Hugging Face-backed model spec.

pub fn from_source( name: impl Into<String>, task: ModelTask, source: ModelSource, ) -> Self

Creates a model spec from explicit source.

pub fn repo_id_value(&self) -> Option<&str>

Returns the Hugging Face repo id when this spec has one.

pub fn revision_value(&self) -> Option<&str>

Returns the revision when this spec has one.

pub fn safe_name(&self) -> String

Returns a filesystem-safe model name segment.

pub fn from_preset(preset: ModelPreset) -> Self

Builds this value from preset.

pub fn name(self, value: impl Into<String>) -> Self

Returns name.

pub fn revision(self, value: impl Into<String>) -> Self

Returns revision.

pub fn file(self, path: impl Into<String>) -> Self

Returns file.

pub fn optional_file(self, path: impl Into<String>) -> Self

Returns optional file.

pub fn first_available_file( self, paths: impl IntoIterator<Item = impl Into<String>>, ) -> Self

Returns first available file.

pub fn cuda_oxide_plan( self, module_name: impl Into<String>, kernel_names: impl IntoIterator<Item = impl Into<String>>, ) -> CudaOxideModelPlan

Builds a cuda-oxide runtime plan for this model spec.

Trait Implementations

impl Clone for ModelSpec

fn clone(&self) -> ModelSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ModelSpec

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for ModelSpec

impl PartialEq for ModelSpec

fn eq(&self, other: &ModelSpec) -> 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 Serialize for ModelSpec

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for ModelSpec