From b1a6f228cb8072d8926ba399c6d6fc07fdf060de Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Thu, 11 May 2023 10:54:31 -0700 Subject: [PATCH] make style --- src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py | 2 +- tests/pipelines/unidiffuser/test_unidiffuser.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py b/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py index efd1a15be8..e600429c7e 100644 --- a/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py +++ b/src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py @@ -160,7 +160,7 @@ class UniDiffuserTextDecoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): def get_dummy_token(self, batch_size: int, device: torch.device) -> torch.Tensor: return torch.zeros(batch_size, self.prefix_length, dtype=torch.int64, device=device) - + def encode(self, prefix): return self.encode_prefix(prefix) diff --git a/tests/pipelines/unidiffuser/test_unidiffuser.py b/tests/pipelines/unidiffuser/test_unidiffuser.py index 967284876a..3fb9d9e9b5 100644 --- a/tests/pipelines/unidiffuser/test_unidiffuser.py +++ b/tests/pipelines/unidiffuser/test_unidiffuser.py @@ -454,7 +454,7 @@ class UniDiffuserPipelineFastTests(PipelineTesterMixin, unittest.TestCase): image_slice = image[0, -3:, -3:, -1] expected_img_slice = np.array([0.5757, 0.6270, 0.6567, 0.4966, 0.4639, 0.5664, 0.5259, 0.5068, 0.5713]) assert np.abs(image_slice.flatten() - expected_img_slice).max() < 1e-3 - + @require_torch_gpu def test_unidiffuser_default_img2text_v1_cuda_fp16(self): device = "cuda"