increase stable diffusion validation threshold 1e-4 -> 3e-4 (#2897)

saw a flaky CI failure with 1.1e-4, and 3e-4 is a good number
2023-12-21 11:45:25 -05:00
parent 24e79e0f53
commit 7dc3352877
1 changed files with 1 additions and 1 deletions
@@ -649,5 +649,5 @@ if __name__ == "__main__":
  if args.prompt == default_prompt and args.steps == 5 and args.seed == 0 and args.guidance == 7.5:
    ref_image = Tensor(np.array(Image.open(Path(__file__).parent / "stable_diffusion_seed0.png")))
    distance = (((x - ref_image).cast(dtypes.float) / ref_image.max())**2).mean().item()
-    assert distance < 1e-4, f"validation failed with {distance=}"
+    assert distance < 3e-4, f"validation failed with {distance=}"
    print(colored(f"output validated with {distance=}", "green"))