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

updated TransformerModel init

Browse Source 
tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")
This commit is contained in:
Jeff Vestal
2023-12-13 09:26:13 -06:00
committed by GitHub
parent dab4e0f46f
commit c8fedbb407
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": [
"# Set the model name from Hugging Face and task type\n",
"hf_model_id='sentence-transformers/all-distilroberta-v1'\n",
"tm = TransformerModel(hf_model_id, \"text_embedding\")\n",
"tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")\n",
"\n",
"#set the modelID as it is named in Elasticsearch\n",
"es_model_id = tm.elasticsearch_model_id()\n",