aljaz/ElasticDocs_GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/ElasticDocs_GPT.git synced 2025-12-17 05:44:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update load_embedding_model.ipynb

Browse Source
This commit is contained in:
Jeff Vestal
2023-12-13 09:27:01 -06:00
committed by GitHub
parent c8fedbb407
commit c369d5e51c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
load_embedding_model.ipynb
View File

@@ -166,7 +166,7 @@
"source": [ "source": [
"# Set the model name from Hugging Face and task type\n", "# Set the model name from Hugging Face and task type\n",
"hf_model_id='sentence-transformers/all-distilroberta-v1'\n", "hf_model_id='sentence-transformers/all-distilroberta-v1'\n",
"tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")\n", "tm = TransformerModel(model_id=hf_model_id, task_type='text_embedding')\n",
"\n", "\n",
"#set the modelID as it is named in Elasticsearch\n", "#set the modelID as it is named in Elasticsearch\n",
"es_model_id = tm.elasticsearch_model_id()\n", "es_model_id = tm.elasticsearch_model_id()\n",