AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Make style and remove some testing code.

Browse Source
This commit is contained in:
Daniel Gu
2023-05-11 16:32:20 -07:00
parent 54cfa3b370
commit 10e3774b8e
2 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py
View File

@@ -1317,7 +1317,7 @@ class UniDiffuserPipeline(DiffusionPipeline):
latents = self._combine(image_vae_latents, image_clip_latents)
elif mode in ["img2text", "text"]:
latents = prompt_embeds
print(f"Initial latents: {latents}")
# 7. Check that shapes of latents and image match the UNet channels.

11
tests/pipelines/unidiffuser/test_unidiffuser.py
View File

@@ -589,10 +589,9 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase):
expected_text_prefix = "Astronaut "
assert text[0][:10] == expected_text_prefix
def test_unidiffuser_default_joint_v1_fp16(self):
pipe = UniDiffuserPipeline.from_pretrained("dg845/unidiffuser-diffusers", torch_dtype=torch.float16)
pipe.image_encoder = CLIPVisionModelWithProjection.from_pretrained("/home/tamamo/scratch/unidiffuser_big/image_encoder")
pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None)
pipe.enable_attention_slicing()
@@ -613,10 +612,9 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase):
expected_text_prefix = "Pink pink "
assert text[0][:10] == expected_text_prefix
def test_unidiffuser_default_text2img_v1_fp16(self):
pipe = UniDiffuserPipeline.from_pretrained("dg845/unidiffuser-diffusers", torch_dtype=torch.float16)
pipe.image_encoder = CLIPVisionModelWithProjection.from_pretrained("/home/tamamo/scratch/unidiffuser_big/image_encoder")
pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None)
pipe.enable_attention_slicing()
@@ -630,10 +628,9 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase):
image_slice = image[0, -3:, -3:, -1]
expected_slice = np.array([0.4702, 0.4666, 0.4446, 0.4829, 0.4468, 0.4565, 0.4663, 0.4956, 0.4277])
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
def test_unidiffuser_default_text2img_v1_fp16_no_cfg(self):
pipe = UniDiffuserPipeline.from_pretrained("dg845/unidiffuser-diffusers", torch_dtype=torch.float16)
pipe.image_encoder = CLIPVisionModelWithProjection.from_pretrained("/home/tamamo/scratch/unidiffuser_big/image_encoder")
pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None)
pipe.enable_attention_slicing()
@@ -651,7 +648,6 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase):
def test_unidiffuser_default_img2text_v1_fp16(self):
pipe = UniDiffuserPipeline.from_pretrained("dg845/unidiffuser-diffusers", torch_dtype=torch.float16)
pipe.image_encoder = CLIPVisionModelWithProjection.from_pretrained("/home/tamamo/scratch/unidiffuser_big/image_encoder")
pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None)
pipe.enable_attention_slicing()
@@ -663,4 +659,3 @@ class UniDiffuserPipelineSlowTests(unittest.TestCase):
expected_text_prefix = "Astronaut "
assert text[0][:10] == expected_text_prefix