Skip to content

GitLab

Unverified Commit c10861ee authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix test_float16_inference (#4412) 



fix
Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent 94b332c4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/pipelines/test_pipelines_common.py
View file @ c10861ee
...@@ -528,6 +528,7 @@ class PipelineTesterMixin: ...@@ -528,6 +528,7 @@ class PipelineTesterMixin:
pipe.to(torch_device) pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None) pipe.set_progress_bar_config(disable=None)
components = self.get_dummy_components()
pipe_fp16 = self.pipeline_class(**components) pipe_fp16 = self.pipeline_class(**components)
pipe_fp16.to(torch_device, torch.float16) pipe_fp16.to(torch_device, torch.float16)
pipe_fp16.set_progress_bar_config(disable=None) pipe_fp16.set_progress_bar_config(disable=None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment