From c10861ee1b2e9faa37d052e02f48cced5866f87a Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Tue, 1 Aug 2023 07:49:50 -1000 Subject: [PATCH] fix test_float16_inference (#4412) fix Co-authored-by: yiyixuxu --- tests/pipelines/test_pipelines_common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py index 1c71e2a908..d9a2e04485 100644 --- 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)