diff --git a/examples/stable_diffusion.py b/examples/stable_diffusion.py index 51b8748056..72e806f41d 100644 --- a/examples/stable_diffusion.py +++ b/examples/stable_diffusion.py @@ -290,5 +290,5 @@ if __name__ == "__main__": if args.prompt == default_prompt and args.steps == 6 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.cast(dtypes.float) - ref_image.cast(dtypes.float)) / ref_image.max())**2).mean().item() - assert distance < 45e-5, colored(f"validation failed with {distance=}", "red") + assert distance < 50e-5, colored(f"validation failed with {distance=}", "red") print(colored(f"output validated with {distance=}", "green"))