We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a5367cf commit 743e58aCopy full SHA for 743e58a
tests/pipelines/deepfloyd_if/__init__.py
@@ -252,6 +252,9 @@ def _test_save_load_optional_components(self):
252
# This should be handled in the base test and then this method can be removed.
253
def _test_save_load_local(self):
254
components = self.get_dummy_components()
255
+ for key in components:
256
+ if "text_encoder" in key and hasattr(components[key], "eval"):
257
+ components[key].eval()
258
pipe = self.pipeline_class(**components)
259
pipe.to(torch_device)
260
pipe.set_progress_bar_config(disable=None)
0 commit comments