Skip to content

Commit a5367cf

Browse files
committed
fix
1 parent 4425f61 commit a5367cf

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/pipelines/test_pipelines_common.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2101,6 +2101,9 @@ def test_encode_prompt_works_in_isolation(self, extra_required_param_value_dict=
21012101
return
21022102

21032103
components = self.get_dummy_components()
2104+
for key in components:
2105+
if "text_encoder" in key and hasattr(components[key], "eval"):
2106+
components[key].eval()
21042107

21052108
# We initialize the pipeline with only text encoders and tokenizers,
21062109
# mimicking a real-world scenario.

0 commit comments

Comments
 (0)