Enum rust_bert::pipelines::common::ModelResource

pub enum ModelResource {
    Torch(Box<dyn ResourceProvider + Send>),
}

Variants to store either a Torch model resource or ONNX resources

Variants

Torch(Box<dyn ResourceProvider + Send>)

Implementations

impl ModelResource

pub fn get_torch_local_path(&self) -> Result<PathBuf, RustBertError>

Provides the torch resource local path. Returns an error if the variant is not a ModelResources::TORCH

Trait Implementations

impl Debug for ModelResource

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

Formats the value using the given formatter.

impl ResourceProvider for ModelResource

fn get_local_path(&self) -> Result<PathBuf, RustBertError>

Provides the local path for a resource.

fn get_resource(&self) -> Result<Resource<'_>, RustBertError>

Provides access to an underlying resource.