Skip to content

Commit 743e58a

Browse files
committed
more fixes
1 parent a5367cf commit 743e58a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/pipelines/deepfloyd_if/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -252,6 +252,9 @@ def _test_save_load_optional_components(self):
252252
# This should be handled in the base test and then this method can be removed.
253253
def _test_save_load_local(self):
254254
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()
255258
pipe = self.pipeline_class(**components)
256259
pipe.to(torch_device)
257260
pipe.set_progress_bar_config(disable=None)

0 commit comments

Comments
 (0)