Enum NliConfig 

pub enum NliConfig {
    HuggingFace {
        repo: String,
        model_file: String,
        tokenizer_file: String,
    },
}

Source of the zero-shot NLI classifier model.

Selects which model NliClassifier::new downloads and loads for the categorize stage. NliConfig::default is the model memoir ships with (MoritzLaurer’s DeBERTa-v3-xsmall); a consumer overrides it to point the classifier at a different HuggingFace repo. Mirrors crate::llm::LlmConfig’s enum-of-sources shape — one variant today, room to add others (a local path, a direct URL) without a breaking change.

Examples

// The shipped default.
let config = NliConfig::default();

// A different zero-shot NLI repo, ONNX export + tokenizer.
let custom = NliConfig::huggingface(
    "my-org/my-zeroshot-model",
    "onnx/model.onnx",
    "tokenizer.json",
);

Variants

HuggingFace

A model hosted on the HuggingFace Hub.

Fields

repo: String

Repo id, e.g. "MoritzLaurer/deberta-v3-xsmall-zeroshot-v1.1-all-33".

model_file: String

Path to the ONNX model file within the repo.

tokenizer_file: String

Path to the tokenizer JSON within the repo.

Implementations

impl NliConfig

pub fn huggingface( repo: impl Into<String>, model_file: impl Into<String>, tokenizer_file: impl Into<String>, ) -> Self

Builds a config for a model on the HuggingFace Hub.

Trait Implementations

impl Clone for NliConfig

fn clone(&self) -> NliConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NliConfig

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

Formats the value using the given formatter.

impl Default for NliConfig

fn default() -> Self

The model memoir ships with — MoritzLaurer’s DeBERTa-v3-xsmall.

impl Eq for NliConfig

impl Hash for NliConfig

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for NliConfig

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