remove ._static_mode for static model (PaddlePaddle#8214)

gongel · Mar 29, 2024 · 4d661bc · 4d661bc
1 parent 50b4491
commit 4d661bc
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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,