fix test_float16_inference (#4412)

fix Co-authored-by: yiyixuxu <yixu310@gmail,com>

fix test_float16_inference (#4412)
fix Co-authored-by: yiyixuxu <yixu310@gmail,com>
c10861ee · YiYi Xu · GitHub · 94b332c4 · c10861ee
Unverified Commit c10861ee authored Aug 01, 2023 by YiYi Xu Committed by GitHub Aug 01, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/pipelines/test_pipelines_common.py tests/pipelines/test_pipelines_common.py +1 -0

No files found.
--- a/tests/pipelines/test_pipelines_common.py
+++ b/tests/pipelines/test_pipelines_common.py
@@ -528,6 +528,7 @@ class PipelineTesterMixin:
        pipe.to(torch_device)
        pipe.set_progress_bar_config(disable=None)

+        components = self.get_dummy_components()
        pipe_fp16 = self.pipeline_class(**components)
        pipe_fp16.to(torch_device, torch.float16)
        pipe_fp16.set_progress_bar_config(disable=None)