Skip to content

Commit ba35924

Browse files
committed
add one more
1 parent 9802231 commit ba35924

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -377,9 +377,10 @@ def test_text_to_image_face_id(self):
377377
pipeline.set_ip_adapter_scale(0.7)
378378

379379
inputs = self.get_dummy_inputs()
380-
id_embeds = load_pt("https://huggingface.co/datasets/fabiorigano/testing-images/resolve/main/ai_face2.ipadpt")[
381-
0
382-
]
380+
id_embeds = load_pt(
381+
"https://huggingface.co/datasets/fabiorigano/testing-images/resolve/main/ai_face2.ipadpt",
382+
map_location=torch_device,
383+
)[0]
383384
id_embeds = id_embeds.reshape((2, 1, 1, 512))
384385
inputs["ip_adapter_image_embeds"] = [id_embeds]
385386
inputs["ip_adapter_image"] = None

0 commit comments

Comments
 (0)