Skip to content

Commit

Permalink
rollback to 50 inference steps in IP-Adapter tests
Browse files Browse the repository at this point in the history
Follow up of 8a36c8c

This is what is used in the official notebook (ip_adapter_demo.ipynb and
ip_adapter-plus_demo.ipynb)
  • Loading branch information
deltheil committed Jan 22, 2024
1 parent ed36213 commit 1e7ef74
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/e2e/test_diffusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -1236,7 +1236,7 @@ def test_diffusion_ip_adapter(
clip_image_embedding = ip_adapter.compute_clip_image_embedding(ip_adapter.preprocess_image(woman_image))
ip_adapter.set_clip_image_embedding(clip_image_embedding)

sd15.set_inference_steps(30)
sd15.set_inference_steps(50)

manual_seed(2)
x = torch.randn(1, 4, 64, 64, device=test_device, dtype=torch.float16)
Expand Down Expand Up @@ -1382,7 +1382,7 @@ def test_diffusion_ip_adapter_plus(
clip_image_embedding = ip_adapter.compute_clip_image_embedding(ip_adapter.preprocess_image(statue_image))
ip_adapter.set_clip_image_embedding(clip_image_embedding)

sd15.set_inference_steps(30)
sd15.set_inference_steps(50)

manual_seed(42) # seed=42 is used in the official IP-Adapter demo
x = torch.randn(1, 4, 64, 64, device=test_device, dtype=torch.float16)
Expand Down

0 comments on commit 1e7ef74

Please sign in to comment.