summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-02-16 09:57:55 +0200
committerGeorgi Gerganov <ggerganov@gmail.com>2024-02-16 09:57:55 +0200
commit594845aab1c6775877f6d9545a51dc0f8d0b3d77 (patch)
treeff485a93e1a887fc7db20fc98955d62ed525420a
parent4524290e87b8e107cc2b56e1251751546f4b9051 (diff)
ci : fix BERT model download and convert
-rwxr-xr-xci/run.sh4
1 files changed, 4 insertions, 0 deletions
diff --git a/ci/run.sh b/ci/run.sh
index a4264d77..979b4a79 100755
--- a/ci/run.sh
+++ b/ci/run.sh
@@ -580,6 +580,10 @@ function gg_run_embd_bge_small {
gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/resolve/main/pytorch_model.bin
gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/sentence_bert_config.json
gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/vocab.txt
+ gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/modules.json
+ gg_wget models-mnt/bge-small/ https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/config.json
+
+ gg_wget models-mnt/bge-small/1_Pooling https://huggingface.co/BAAI/bge-small-en-v1.5/raw/main/1_Pooling/config.json
path_models="../models-mnt/bge-small"