up

Author: sayakpaul

tests/pipelines/test_pipelines_common.py

@@ -1563,6 +1563,9 @@ def _test_attention_slicing_forward_pass(
             return
 
         components = self.get_dummy_components()
+        for key in components:
+            if "text_encoder" in key and hasattr(components[key], "eval"):
+                components[key].eval()
         pipe = self.pipeline_class(**components)
         for component in pipe.components.values():
             if hasattr(component, "set_default_attn_processor"):