Skip to content

Commit a4d9d5b

Browse files
committed
Remove default --model-id and udpate docstring
1 parent 6e900af commit a4d9d5b

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

router/src/main.rs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,14 @@ static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
1212
#[derive(Parser, Redact)]
1313
#[clap(author, version, about, long_about = None)]
1414
struct Args {
15-
/// The name of the model to load.
16-
/// Can be a MODEL_ID as listed on <https://hf.co/models> like
17-
/// `BAAI/bge-large-en-v1.5`.
18-
/// Or it can be a local directory containing the necessary files
19-
/// as saved by `save_pretrained(...)` methods of transformers
20-
#[clap(default_value = "BAAI/bge-large-en-v1.5", long, env)]
15+
/// The Hugging Face model ID, can be any model listed on <https://huggingface.co/models> with
16+
/// the `text-embeddings-inference` tag (meaning it's compatible with Text Embeddings
17+
/// Inference)
18+
///
19+
/// Alternatively, the specified ID can also be a path to a local directory containing the
20+
/// necessary model files saved by the `save_pretrained(...)` methods of either Transformers or
21+
/// Sentence Transformers.
22+
#[clap(long, env)]
2123
#[redact(partial)]
2224
model_id: String,
2325

0 commit comments

Comments
 (0)