From 8f2d325c2d823628759be2f022a99ba526a0cd88 Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Wed, 10 May 2023 23:53:40 -0700 Subject: [PATCH] make style --- tests/pipelines/unidiffuser/test_unidiffuser.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/pipelines/unidiffuser/test_unidiffuser.py b/tests/pipelines/unidiffuser/test_unidiffuser.py index 74aba2a165..281419469d 100644 --- a/tests/pipelines/unidiffuser/test_unidiffuser.py +++ b/tests/pipelines/unidiffuser/test_unidiffuser.py @@ -3,7 +3,6 @@ import random import unittest import numpy as np -import pytest import torch from PIL import Image from transformers import ( @@ -440,7 +439,7 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase): # TODO: get correct image slice expected_slice = np.array([0.3965, 0.4568, 0.4495, 0.4590, 0.4465, 0.4690, 0.5454, 0.5093, 0.4321]) assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3 - + def test_unidiffuser_default_img2text(self): pipe = UniDiffuserPipeline.from_pretrained("dg845/unidiffuser-diffusers") pipe.to(torch_device) @@ -455,5 +454,3 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase): # TODO: get correct text prefix expected_text_prefix = " no no no " assert text[0][:10] == expected_text_prefix - -