forked from lukaszett/Knack-Scraper
Use different embeddings model;
This commit is contained in:
parent
49239e7e25
commit
8fae350b34
10 changed files with 1846 additions and 57 deletions
|
|
@ -241,7 +241,7 @@ def create_default_pipeline(device: str = "cpu",
|
|||
node_class=TextEmbeddingNode,
|
||||
node_kwargs={
|
||||
'device': device,
|
||||
'model_path': os.environ.get('MINILM_MODEL_PATH')
|
||||
'model_path': os.environ.get('GTE_MODEL_PATH')
|
||||
},
|
||||
dependencies=[],
|
||||
name='TextEmbeddingNode'
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue