File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -12,12 +12,14 @@ static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
12
12
#[ derive( Parser , Redact ) ]
13
13
#[ clap( author, version, about, long_about = None ) ]
14
14
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) ]
21
23
#[ redact( partial) ]
22
24
model_id : String ,
23
25
You can’t perform that action at this time.
0 commit comments