From a4d9d5b507194b0f9169be484d2045fc1ba0a440 Mon Sep 17 00:00:00 2001 From: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com> Date: Wed, 9 Jul 2025 09:10:51 +0200 Subject: [PATCH] Remove default `--model-id` and udpate docstring --- router/src/main.rs | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/router/src/main.rs b/router/src/main.rs index 39b975d5..4decef83 100644 --- a/router/src/main.rs +++ b/router/src/main.rs @@ -12,12 +12,14 @@ static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc; #[derive(Parser, Redact)] #[clap(author, version, about, long_about = None)] struct Args { - /// The name of the model to load. - /// Can be a MODEL_ID as listed on like - /// `BAAI/bge-large-en-v1.5`. - /// Or it can be a local directory containing the necessary files - /// as saved by `save_pretrained(...)` methods of transformers - #[clap(default_value = "BAAI/bge-large-en-v1.5", long, env)] + /// The Hugging Face model ID, can be any model listed on with + /// the `text-embeddings-inference` tag (meaning it's compatible with Text Embeddings + /// Inference) + /// + /// Alternatively, the specified ID can also be a path to a local directory containing the + /// necessary model files saved by the `save_pretrained(...)` methods of either Transformers or + /// Sentence Transformers. + #[clap(long, env)] #[redact(partial)] model_id: String,