diff --git a/tests/expected_output/test_imagine[k_dpm_2_a]_.png b/tests/expected_output/test_imagine[k_dpm_2_a]_.png
index ea8b551..b70279d 100644
Binary files a/tests/expected_output/test_imagine[k_dpm_2_a]_.png and b/tests/expected_output/test_imagine[k_dpm_2_a]_.png differ
diff --git a/tests/expected_output/test_imagine[k_euler_a]_.png b/tests/expected_output/test_imagine[k_euler_a]_.png
index 65b42f0..c5fce1d 100644
Binary files a/tests/expected_output/test_imagine[k_euler_a]_.png and b/tests/expected_output/test_imagine[k_euler_a]_.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 681d1ba..fbd79a8 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_euler_a]_.png b/tests/expected_output/test_img2img_beach_to_sunset[k_euler_a]_.png
index 772e2b8..f3fe717 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_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 674f210..482a5f7 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_euler_a]_.png b/tests/expected_output/test_img_to_img_from_url_cats[k_euler_a]_.png
index 1ee41a6..fc98e0a 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_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 a1d0686..f2b662d 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 6a144cf..a9ae4fa 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 2da4f0c..b3c348a 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_euler_a-0.05]_.png b/tests/expected_output/test_img_to_img_fruit_2_gold[k_euler_a-0.05]_.png
index cad2fde..ca20a34 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 9dc4a63..3cb720c 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 e64a4f4..b161b6d 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_model_versions__SD-1.4.png b/tests/expected_output/test_model_versions__SD-1.4.png
deleted file mode 100644
index 5c21e34..0000000
Binary files a/tests/expected_output/test_model_versions__SD-1.4.png and /dev/null differ
diff --git a/tests/expected_output/test_model_versions__SD-1.5.png b/tests/expected_output/test_model_versions__SD-1.5.png
deleted file mode 100644
index 1923b21..0000000
Binary files a/tests/expected_output/test_model_versions__SD-1.5.png and /dev/null differ
diff --git a/tests/expected_output/test_model_versions__SD-2.0-v.png b/tests/expected_output/test_model_versions__SD-2.0-v.png
deleted file mode 100644
index ce67cdc..0000000
Binary files a/tests/expected_output/test_model_versions__SD-2.0-v.png and /dev/null differ
diff --git a/tests/expected_output/test_model_versions__SD-2.0.png b/tests/expected_output/test_model_versions__SD-2.0.png
deleted file mode 100644
index d3e2160..0000000
Binary files a/tests/expected_output/test_model_versions__SD-2.0.png and /dev/null differ
diff --git a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.4.png b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.4.png
new file mode 100644
index 0000000..db4b200
Binary files /dev/null and b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.4.png differ
diff --git a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.5.png b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.5.png
new file mode 100644
index 0000000..ffa6acb
Binary files /dev/null and b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-1.5.png differ
diff --git a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0-v.png b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0-v.png
new file mode 100644
index 0000000..1d3818c
Binary files /dev/null and b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0-v.png differ
diff --git a/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0.png b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0.png
new file mode 100644
index 0000000..b012701
Binary files /dev/null and b/tests/expected_output/test_model_versions__a headshot photo of a happy couple smiling at the camera_SD-2.0.png differ
diff --git a/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.4.png b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.4.png
new file mode 100644
index 0000000..19c8008
Binary files /dev/null and b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.4.png differ
diff --git a/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.5.png b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.5.png
new file mode 100644
index 0000000..b364750
Binary files /dev/null and b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-1.5.png differ
diff --git a/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0-v.png b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0-v.png
new file mode 100644
index 0000000..5ee3197
Binary files /dev/null and b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0-v.png differ
diff --git a/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0.png b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0.png
new file mode 100644
index 0000000..3bafb0e
Binary files /dev/null and b/tests/expected_output/test_model_versions__a painting of a beautiful cloudy sunset at the beach_SD-2.0.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.4.png b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.4.png
new file mode 100644
index 0000000..9ff4c3e
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.4.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.5.png b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.5.png
new file mode 100644
index 0000000..b397eba
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-1.5.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0-v.png b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0-v.png
new file mode 100644
index 0000000..8eaecf0
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0-v.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0.png b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0.png
new file mode 100644
index 0000000..060b162
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a bowl of fruit_SD-2.0.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a dog_SD-1.4.png b/tests/expected_output/test_model_versions__a photo of a dog_SD-1.4.png
new file mode 100644
index 0000000..07206da
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a dog_SD-1.4.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a dog_SD-1.5.png b/tests/expected_output/test_model_versions__a photo of a dog_SD-1.5.png
new file mode 100644
index 0000000..44c3c5b
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a dog_SD-1.5.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0-v.png b/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0-v.png
new file mode 100644
index 0000000..4afc0d3
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0-v.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0.png b/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0.png
new file mode 100644
index 0000000..e6be7f4
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a dog_SD-2.0.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.4.png b/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.4.png
new file mode 100644
index 0000000..04145eb
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.4.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.5.png b/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.5.png
new file mode 100644
index 0000000..948cc16
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a handshake_SD-1.5.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0-v.png b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0-v.png
new file mode 100644
index 0000000..9f686b4
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0-v.png differ
diff --git a/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0.png b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0.png
new file mode 100644
index 0000000..b9644cb
Binary files /dev/null and b/tests/expected_output/test_model_versions__a photo of a handshake_SD-2.0.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-1.4.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-1.4.png
new file mode 100644
index 0000000..ead591c
Binary files /dev/null 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-1.4.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-1.5.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-1.5.png
new file mode 100644
index 0000000..9f26d20
Binary files /dev/null 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-1.5.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.0-v.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.0-v.png
new file mode 100644
index 0000000..08a204c
Binary files /dev/null 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.0-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.0.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.0.png
new file mode 100644
index 0000000..728e7d2
Binary files /dev/null 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.0.png differ