diff --git a/tests/expected_output/test_clip_masking_.png b/tests/expected_output/test_clip_masking_.png
index 542d242..4e4240d 100644
Binary files a/tests/expected_output/test_clip_masking_.png and b/tests/expected_output/test_clip_masking_.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+101_bin.png b/tests/expected_output/test_clip_masking__mask+101_bin.png
index aa91de0..f6c483d 100644
Binary files a/tests/expected_output/test_clip_masking__mask+101_bin.png and b/tests/expected_output/test_clip_masking__mask+101_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+101_g.png b/tests/expected_output/test_clip_masking__mask+101_g.png
index 2d7d51f..db39307 100644
Binary files a/tests/expected_output/test_clip_masking__mask+101_g.png and b/tests/expected_output/test_clip_masking__mask+101_g.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+11_bin.png b/tests/expected_output/test_clip_masking__mask+11_bin.png
index e960561..6603376 100644
Binary files a/tests/expected_output/test_clip_masking__mask+11_bin.png and b/tests/expected_output/test_clip_masking__mask+11_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+11_g.png b/tests/expected_output/test_clip_masking__mask+11_g.png
index cf0da6f..ce247cc 100644
Binary files a/tests/expected_output/test_clip_masking__mask+11_g.png and b/tests/expected_output/test_clip_masking__mask+11_g.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+1_bin.png b/tests/expected_output/test_clip_masking__mask+1_bin.png
index ccaaac5..cc4e077 100644
Binary files a/tests/expected_output/test_clip_masking__mask+1_bin.png and b/tests/expected_output/test_clip_masking__mask+1_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__mask+1_g.png b/tests/expected_output/test_clip_masking__mask+1_g.png
index 054a805..9823f7b 100644
Binary files a/tests/expected_output/test_clip_masking__mask+1_g.png and b/tests/expected_output/test_clip_masking__mask+1_g.png differ
diff --git a/tests/expected_output/test_clip_masking__mask-25_bin.png b/tests/expected_output/test_clip_masking__mask-25_bin.png
index 56fcb7c..8c9f28c 100644
Binary files a/tests/expected_output/test_clip_masking__mask-25_bin.png and b/tests/expected_output/test_clip_masking__mask-25_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__mask-25_g.png b/tests/expected_output/test_clip_masking__mask-25_g.png
index 76a0bc7..3c9ef08 100644
Binary files a/tests/expected_output/test_clip_masking__mask-25_g.png and b/tests/expected_output/test_clip_masking__mask-25_g.png differ
diff --git a/tests/expected_output/test_clip_masking__maskx0.5_bin.png b/tests/expected_output/test_clip_masking__maskx0.5_bin.png
index 56fcb7c..8c9f28c 100644
Binary files a/tests/expected_output/test_clip_masking__maskx0.5_bin.png and b/tests/expected_output/test_clip_masking__maskx0.5_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__maskx0.5_g.png b/tests/expected_output/test_clip_masking__maskx0.5_g.png
index 4676699..d5e126f 100644
Binary files a/tests/expected_output/test_clip_masking__maskx0.5_g.png and b/tests/expected_output/test_clip_masking__maskx0.5_g.png differ
diff --git a/tests/expected_output/test_clip_masking__maskx6_bin.png b/tests/expected_output/test_clip_masking__maskx6_bin.png
index edc52d4..fdd661a 100644
Binary files a/tests/expected_output/test_clip_masking__maskx6_bin.png and b/tests/expected_output/test_clip_masking__maskx6_bin.png differ
diff --git a/tests/expected_output/test_clip_masking__maskx6_g.png b/tests/expected_output/test_clip_masking__maskx6_g.png
index 6bebb04..feab377 100644
Binary files a/tests/expected_output/test_clip_masking__maskx6_g.png and b/tests/expected_output/test_clip_masking__maskx6_g.png differ
diff --git a/tests/expected_output/test_cliptext_inpainting_pearl_doctor_.png b/tests/expected_output/test_cliptext_inpainting_pearl_doctor_.png
index 0ef3442..159c796 100644
Binary files a/tests/expected_output/test_cliptext_inpainting_pearl_doctor_.png and b/tests/expected_output/test_cliptext_inpainting_pearl_doctor_.png differ
diff --git a/tests/expected_output/test_imagine[k_lms]_.png b/tests/expected_output/test_imagine[k_lms]_.png
index c749164..0480cdb 100644
Binary files a/tests/expected_output/test_imagine[k_lms]_.png and b/tests/expected_output/test_imagine[k_lms]_.png differ
diff --git a/tests/expected_output/test_imagine[plms]_.png b/tests/expected_output/test_imagine[plms]_.png
index 254c139..91e34c3 100644
Binary files a/tests/expected_output/test_imagine[plms]_.png and b/tests/expected_output/test_imagine[plms]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[ddim]_.png b/tests/expected_output/test_img2img_beach_to_sunset[ddim]_.png
index e9e9ca4..641102d 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[ddim]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[ddim]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2]_.png
index 4af6431..e3a8eb7 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2_a]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2_a]_.png
index da06470..5f447ba 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2_a]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_2_a]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_adaptive]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_adaptive]_.png
index 833c118..72d60ea 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_adaptive]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_adaptive]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_fast]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_fast]_.png
index 9de74c0..0c19f08 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_fast]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpm_fast]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2m]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2m]_.png
index 647e3ab..fb5b38c 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2m]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2m]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2s_a]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2s_a]_.png
index 2017d34..31c0a48 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2s_a]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_dpmpp_2s_a]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_euler]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_euler]_.png
index a55889d..3f2b3e5 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_euler]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_euler]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_euler_a]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_euler_a]_.png
index eccd716..da0f150 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_euler_a]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_euler_a]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_heun]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_heun]_.png
index d5494cf..487cbbb 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_heun]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_heun]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[k_lms]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_lms]_.png
index 986880d..e91be31 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[k_lms]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[k_lms]_.png differ
diff --git a/tests/expected_output/test_img2img_beach_to_sunset[plms]_.png b/tests/expected_output/test_img2img_beach_to_sunset[plms]_.png
index 8731eba..a5d4e04 100644
Binary files a/tests/expected_output/test_img2img_beach_to_sunset[plms]_.png and b/tests/expected_output/test_img2img_beach_to_sunset[plms]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[ddim]_.png b/tests/expected_output/test_img_to_img_from_url_cats[ddim]_.png
index f8f33ad..7b5f410 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[ddim]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[ddim]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2]_.png
index 5eaf52f..ae9367f 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2_a]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2_a]_.png
index ddea261..58e907a 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2_a]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_2_a]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_adaptive]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_adaptive]_.png
index 4420960..fddcb70 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_adaptive]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_adaptive]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_fast]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_fast]_.png
index bfd90d3..d72e019 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_fast]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpm_fast]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2m]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2m]_.png
index 38f30ca..de34e4a 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2m]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2m]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2s_a]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2s_a]_.png
index fb7a0b9..65c148e 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2s_a]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_dpmpp_2s_a]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_euler]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_euler]_.png
index 8bc00bd..81b99cb 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_euler]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_euler]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_euler_a]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_euler_a]_.png
index 47a50bd..913c6ea 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_euler_a]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_euler_a]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_heun]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_heun]_.png
index ddef79f..8defe76 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_heun]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_heun]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[k_lms]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_lms]_.png
index 93a9cfd..4f83aa3 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[k_lms]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[k_lms]_.png differ
diff --git a/tests/expected_output/test_img_to_img_from_url_cats[plms]_.png b/tests/expected_output/test_img_to_img_from_url_cats[plms]_.png
index dafdb75..30c5cf7 100644
Binary files a/tests/expected_output/test_img_to_img_from_url_cats[plms]_.png and b/tests/expected_output/test_img_to_img_from_url_cats[plms]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png
index 2ecb62d..841e850 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png
index 4d4d385..0da49c1 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png
index a6a41e7..8071411 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-1]_.png
index 2d51ae9..5f46666 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[ddim-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png
index 2f95a6a..0b8d241 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png
index aea56db..111f61e 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png
index 1ed115e..65f34f8 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png
index 9e75722..43c4efe 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png
index f9e5185..801e661 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png
index ac37dd4..a3df382 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_2_a-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png
index 7d14724..d6df95a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png
index d46f9ef..952f4dc 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0]_.png
index e5f8114..eb01dc7 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_adaptive-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png
index f669227..51290dc 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png
index 48312dc..286473c 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png
index 08fae84..b041b17 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpm_fast-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png
index ecdf386..818d28d 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png
index 670cf6c..97b4c5f 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png
index 44803da..e051636 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2m-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png
index e6e5f07..5a8a487 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png
index f846d2d..d08c240 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png
index 4c20643..2abb81f 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_dpmpp_2s_a-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png
index 7cb5f82..8ab320e 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png
index 2fcb201..79d48d0 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png
index 660173e..a10f469 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png
index 8b4ccdf..241037b 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png
index f1d9b68..9e25054 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png
index 918351f..c4595a9 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-1]_.png
index f25909d..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png
index a08826f..1219357 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png
index 0379e3e..8ed9e50 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0]_.png
index 946b86f..7e49da5 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_heun-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png
index 01f74d0..0ba07da 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png
index 0b84c7d..ec52001 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png
index 2e69cb5..818af23 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-1]_.png
index fafd4d3..35fca6a 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[k_lms-1]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png
index 09ddcda..e6cabdd 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.05]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png
index 3bbab4d..07d51b0 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0.2]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png
index a265520..c92a9b3 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-0]_.png differ
diff --git a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-1]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-1]_.png
index 2d51ae9..5f46666 100644
Binary files a/tests/expected_output/test_img_to_img_fruit_2_gold[plms-1]_.png and b/tests/expected_output/test_img_to_img_fruit_2_gold[plms-1]_.png differ
diff --git a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.1-v.png b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.1-v.png
index cd3e84e..52458e9 100644
Binary files a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.1-v.png and b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.1-v.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.1-v.png b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.1-v.png
index 3716223..edcb557 100644
Binary files a/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.1-v.png and b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.1-v.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of an astronaut riding a horse on the moon. the earth visible in the background_SD-2.1.png b/tests/expected_output/test_model_versions__a photo of an astronaut riding a horse on the moon. the earth visible in the background_SD-2.1.png
index 5aa9e65..8e2b4aa 100644
Binary files a/tests/expected_output/test_model_versions__a photo of an astronaut riding a horse on the moon. the earth visible in the background_SD-2.1.png and b/tests/expected_output/test_model_versions__a photo of an astronaut riding a horse on the moon. the earth visible in the background_SD-2.1.png differ
diff --git a/tests/expected_output/test_outpainting_outpaint_.png b/tests/expected_output/test_outpainting_outpaint_.png
index a3cb6d3..7a1df0a 100644
Binary files a/tests/expected_output/test_outpainting_outpaint_.png and b/tests/expected_output/test_outpainting_outpaint_.png differ