From 4d661bc34bd21a8150e6c835757a1bfe2477a1ad Mon Sep 17 00:00:00 2001 From: w5688414 Date: Fri, 29 Mar 2024 18:05:46 +0800 Subject: [PATCH] remove ._static_mode for static model (#8214) --- pipelines/pipelines/nodes/retriever/dense.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pipelines/pipelines/nodes/retriever/dense.py b/pipelines/pipelines/nodes/retriever/dense.py index dabb139023e6..4daa2f655cba 100644 --- a/pipelines/pipelines/nodes/retriever/dense.py +++ b/pipelines/pipelines/nodes/retriever/dense.py @@ -163,7 +163,6 @@ def __init__( "feature_extraction", model=query_embedding_model, batch_size=self.batch_size, - _static_mode=True, return_tensors="np", max_len=max_seq_len_query, output_emb_size=output_emb_size, @@ -176,7 +175,6 @@ def __init__( "feature_extraction", model=passage_embedding_model, batch_size=self.batch_size, - _static_mode=True, return_tensors="np", max_len=max_seq_len_passage, output_emb_size=output_emb_size,