bert4vec:一个基于预训练的句向量生成工具 环境
bert4vec:一个基于预训练的句向量生成工具
环境
transformers>=4.6.0,<5.0.0
torch>=1.6.0
numpy
huggingface-hub
faiss (optional)
安装
方式一
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ bert4vec
方式二
git clone https://github.com/zejunwang1/bert4vec
cd bert4vec
转载
2022-02-18 11:07:01 ·
765 阅读 ·
0 评论