shibing624 / text2vec

text2vec, text to vector. 文本向量表征工具,把文本转化为向量矩阵,实现了Word2Vec、RankBM25、Sentence-BERT、CoSENT等文本表征、文本相似度计算模型,开箱即用。
https://pypi.org/project/text2vec/
Apache License 2.0
4.46k stars 396 forks source link

我该如何指定模型所在的位置,这样就不用到Hugging Face下载了 #153

Closed omaiyiwa closed 2 months ago

omaiyiwa commented 2 months ago

Describe the Question

Please provide a clear and concise description of what the question is.

我该如何指定模型所在的位置,这样就不用到Hugging Face下载了 我现在是这样的, self.m = BertSimilarity(model_name_or_path="shibing624/text2vec-base-chinese")

shibing624 commented 2 months ago

"shibing624/text2vec-base-chinese" 改为模型绝对路径。

omaiyiwa commented 2 months ago

谢谢,我发现第一次下载完到本地后,之后如果不指定绝对路径,加载时间是778ms左右,如果加载绝对路径是470ms左右