diff --git a/tests/pipelines/ddim/test_ddim.py b/tests/pipelines/ddim/test_ddim.py index 26ea10caea..df721efcf4 100644 --- a/tests/pipelines/ddim/test_ddim.py +++ b/tests/pipelines/ddim/test_ddim.py @@ -29,6 +29,7 @@ torch.backends.cuda.matmul.allow_tf32 = False class DDIMPipelineFastTests(PipelineTesterMixin, unittest.TestCase): pipeline_class = DDIMPipeline + test_cpu_offload = False def get_dummy_components(self): torch.manual_seed(0) diff --git a/tests/test_pipelines_common.py b/tests/test_pipelines_common.py index 6565a52c2a..af5dd6a403 100644 --- a/tests/test_pipelines_common.py +++ b/tests/test_pipelines_common.py @@ -190,7 +190,7 @@ class PipelineTesterMixin: with tempfile.TemporaryDirectory() as tmpdir: pipe.save_pretrained(tmpdir) - pipe_loaded = self.pipeline_class.from_pretrained(tmpdir) + pipe_loaded = self.pipeline_class.from_pretrained(tmpdir, torch_dtype=torch.float16) pipe_loaded.to(torch_device) pipe_loaded.set_progress_bar_config(disable=None)