Skip to content

Commit 2be151e

Browse files
committed
fix tests
1 parent b780740 commit 2be151e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -774,7 +774,7 @@ def test_stable_diffusion_no_weights_predict():
774774
else:
775775
out = out.numpy()
776776
assert out.shape == (1, 256, 256, 3)
777-
assert out.min() > -8 and out.max() < 8 # It should be within this range
777+
assert out.min() > -10 and out.max() < 10 # It should be within this range
778778

779779

780780
""" Segmentation models """

0 commit comments

Comments
 (0)