fix(scripts): restore MSE calculation, update some test refs

ssube · Jun 8, 2023 · 55614cd · 55614cd
1 parent 7ec2597
commit 55614cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-unipc-0.png b/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-unipc-0.png
diff --git a/api/scripts/test-release.py b/api/scripts/test-release.py
@@ -451,7 +451,7 @@ def run_test(
 
         mse = find_mse(result, ref)
 
-        if mse < (test.mse_threshold * 10):
+        if mse < test.mse_threshold:
             logger.info("MSE within threshold: %.5f < %.5f", mse, test.mse_threshold)
         else:
             logger.warning("MSE above threshold: %.5f > %.5f", mse, test.mse_threshold)