Skip to content

Commit 577fe86

Browse files
committed
Move test
1 parent 9b428b2 commit 577fe86

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

Diff for: tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py

+19
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,25 @@ def test_text_to_image_full_face(self):
201201

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

204+
def test_text_to_image_full_face(self):
205+
image_encoder = self.get_image_encoder(repo_id="h94/IP-Adapter", subfolder="models/image_encoder")
206+
pipeline = StableDiffusionPipeline.from_pretrained(
207+
"runwayml/stable-diffusion-v1-5", image_encoder=image_encoder, safety_checker=None, torch_dtype=self.dtype
208+
)
209+
pipeline.to(torch_device)
210+
pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-full-face_sd15.bin")
211+
pipeline.set_ip_adapter_scale(0.7)
212+
213+
inputs = self.get_dummy_inputs()
214+
images = pipeline(**inputs).images
215+
image_slice = images[0, :3, :3, -1].flatten()
216+
217+
expected_slice = np.array(
218+
[0.1706543, 0.1303711, 0.12573242, 0.21777344, 0.14550781, 0.14038086, 0.40820312, 0.41455078, 0.42529297]
219+
)
220+
221+
assert np.allclose(image_slice, expected_slice, atol=1e-4, rtol=1e-4)
222+
204223

205224
@slow
206225
@require_torch_gpu

0 commit comments

Comments
 (0)