mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

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

Browse source 
saw a flaky CI failure with 1.1e-4, and 3e-4 is a good number
This commit is contained in:
chenyu 2023-12-21 11:45:25 -05:00 • committed by GitHub
commit 7dc3352877
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/stable_diffusion.py
View file

@ -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"))