atrytone commited on
Commit
81222f4
1 Parent(s): a144c9e

Update NBDT_Data_Recs.ipynb

Browse files
Files changed (1) hide show
  1. NBDT_Data_Recs.ipynb +2 -2
NBDT_Data_Recs.ipynb CHANGED
@@ -430,7 +430,7 @@
430
  "name": "stderr",
431
  "output_type": "stream",
432
  "text": [
433
- "WARNING:sentence_transformers.SentenceTransformer:No sentence-transformers model found with name /root/.cache/torch/sentence_transformers/biodatlab_MIReAD-Neuro. Creating a new one with MEAN pooling.\n"
434
  ]
435
  },
436
  {
@@ -770,7 +770,7 @@
770
  "def k_inference(query,k):\n",
771
  " return inference(query,k,'koala')\n",
772
  "\n",
773
- "model_name = \"biodatlab/MIReAD-Neuro\"\n",
774
  "model_kwargs = {'device': 'cpu'}\n",
775
  "encode_kwargs = {'normalize_embeddings': False}\n",
776
  "faiss_embedder = HuggingFaceEmbeddings(\n",
 
430
  "name": "stderr",
431
  "output_type": "stream",
432
  "text": [
433
+ "WARNING:sentence_transformers.SentenceTransformer:No sentence-transformers model found with name /root/.cache/torch/sentence_transformers/biodatlab_MIReAD-Neuro-Large. Creating a new one with MEAN pooling.\n"
434
  ]
435
  },
436
  {
 
770
  "def k_inference(query,k):\n",
771
  " return inference(query,k,'koala')\n",
772
  "\n",
773
+ "model_name = \"biodatlab/MIReAD-Neuro-Large\"\n",
774
  "model_kwargs = {'device': 'cpu'}\n",
775
  "encode_kwargs = {'normalize_embeddings': False}\n",
776
  "faiss_embedder = HuggingFaceEmbeddings(\n",