Skip to content

Commit ed0b471

Browse files
committed
Move test
1 parent f7c42df commit ed0b471

File tree

2 files changed

+19
-94
lines changed

2 files changed

+19
-94
lines changed

tests/pipelines/ip_adapters/test_ip_adapter_full_face_stable_diffusion.py

Lines changed: 0 additions & 94 deletions
This file was deleted.

tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,25 @@ def test_inpainting(self):
152152

153153
assert np.allclose(image_slice, expected_slice, atol=1e-4, rtol=1e-4)
154154

155+
def test_text_to_image_full_face(self):
156+
image_encoder = self.get_image_encoder(repo_id="h94/IP-Adapter", subfolder="models/image_encoder")
157+
pipeline = StableDiffusionPipeline.from_pretrained(
158+
"runwayml/stable-diffusion-v1-5", image_encoder=image_encoder, safety_checker=None, torch_dtype=self.dtype
159+
)
160+
pipeline.to(torch_device)
161+
pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-full-face_sd15.bin")
162+
pipeline.set_ip_adapter_scale(0.7)
163+
164+
inputs = self.get_dummy_inputs()
165+
images = pipeline(**inputs).images
166+
image_slice = images[0, :3, :3, -1].flatten()
167+
168+
expected_slice = np.array(
169+
[0.1706543, 0.1303711, 0.12573242, 0.21777344, 0.14550781, 0.14038086, 0.40820312, 0.41455078, 0.42529297]
170+
)
171+
172+
assert np.allclose(image_slice, expected_slice, atol=1e-4, rtol=1e-4)
173+
155174

156175
@slow
157176
@require_torch_gpu

0 commit comments

Comments
 (0)