Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
51 commits
Select commit Hold shift + click to select a range
c365f4f
update expected results of slow tests
kashif Aug 29, 2022
fa76501
relax sum and mean tests
kashif Aug 29, 2022
21e2897
Print shapes when reporting exception
kashif Aug 29, 2022
b080f71
formatting
kashif Aug 29, 2022
2e7a5d6
fix sentence
kashif Aug 29, 2022
ea46acf
relax test_stable_diffusion_fast_ddim for gpu fp16
kashif Aug 29, 2022
7134264
Merge branch 'main' into fix-tests
kashif Aug 29, 2022
bb6b522
relax flakey tests on GPU
kashif Aug 29, 2022
6e4a075
added comment on large tolerences
kashif Aug 29, 2022
75646db
Merge branch 'main' into fix-tests
kashif Aug 29, 2022
1db4181
Merge branch 'main' into fix-tests
patrickvonplaten Aug 31, 2022
a765a19
black
kashif Aug 31, 2022
71f7683
format
kashif Aug 31, 2022
0a46c6a
set scheduler seed
kashif Aug 31, 2022
5f54c63
Merge branch 'main' into fix-tests
kashif Aug 31, 2022
a1fcd85
added generator
kashif Aug 31, 2022
a0a1d11
use np.isclose
kashif Aug 31, 2022
58199e4
set num_inference_steps to 50
kashif Aug 31, 2022
8b573d2
fix dep. warning
kashif Aug 31, 2022
4c2d1f8
update expected_slice
kashif Aug 31, 2022
af77c11
preprocess if image
kashif Aug 31, 2022
39cff67
updated expected results
kashif Aug 31, 2022
376f8ef
updated expected from CI
kashif Sep 1, 2022
2fe153f
Merge branch 'main' into fix-tests
kashif Sep 1, 2022
3adb0dc
pass generator to VAE
kashif Sep 1, 2022
ac1fe4b
Merge branch 'main' into fix-tests
kashif Sep 1, 2022
f073526
undo change back to orig
kashif Sep 1, 2022
d51eaaf
Merge branch 'main' into fix-tests
kashif Sep 1, 2022
a3f882e
use orignal
kashif Sep 1, 2022
bdcab21
revert back the expected on cpu
kashif Sep 1, 2022
f12359b
Merge branch 'main' into fix-tests
kashif Sep 1, 2022
c193fa7
revert back values for CPU
kashif Sep 1, 2022
3bab624
more undo
kashif Sep 1, 2022
bacd4d8
update result after using gen
kashif Sep 2, 2022
60fa253
update mean
kashif Sep 2, 2022
5a6f65a
Merge branch 'main' into fix-tests
kashif Sep 2, 2022
16649d1
Merge remote-tracking branch 'upstream/main' into fix-tests
kashif Sep 5, 2022
c7736d0
Merge remote-tracking branch 'origin/fix-tests' into fix-tests
kashif Sep 5, 2022
64fc904
Merge branch 'main' into fix-tests
kashif Sep 7, 2022
0e03781
Merge branch 'main' into fix-tests
kashif Sep 9, 2022
8a67749
set generator for mps
kashif Sep 9, 2022
c3532b9
update expected on CI server
kashif Sep 9, 2022
7f1ce10
undo
kashif Sep 9, 2022
e341f48
Merge branch 'main' into fix-tests
kashif Sep 9, 2022
c945094
use new seed every time
kashif Sep 12, 2022
493123b
Merge branch 'main' into fix-tests
kashif Sep 12, 2022
108d588
cpu manual seed
kashif Sep 12, 2022
0fc7ca0
reduce num_inference_steps
kashif Sep 12, 2022
d6d4b27
Merge branch 'main' into fix-tests
kashif Sep 12, 2022
c943f4d
style
kashif Sep 12, 2022
8b84156
use generator for randn
kashif Sep 12, 2022
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion tests/test_pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,7 @@ def test_score_sde_ve_pipeline(self):

assert image.shape == (1, 256, 256, 3)

expected_slice = np.array([0.64363, 0.5868, 0.3031, 0.2284, 0.7409, 0.3216, 0.25643, 0.6557, 0.2633])
expected_slice = np.array([0.35396, 0.31874, 0.0, 0.16337, 0.18218, 0.14668, 0.34573, 0.01166, 0.18164])
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2

@slow
Expand Down
2 changes: 1 addition & 1 deletion tests/test_scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -714,7 +714,7 @@ def test_full_loop_no_noise(self):
result_sum = torch.sum(torch.abs(sample))
result_mean = torch.mean(torch.abs(sample))

assert abs(result_sum.item() - 14379591680.0) < 1e-2
assert abs(result_sum.item() - 14379589632.0) < 1e-2
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Circle CI seemed to be happy with 14379591680.0: https://github.com/huggingface/diffusers/runs/8068605613?check_suite_focus=true#step:6:71 So not sure about this change

=> maybe we should increase the tolerance here?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yes perhaps that is what i was thinking that too so perhaps abs < 2.5 e3 or so

assert abs(result_mean.item() - 18723426.0) < 1e-3

def test_step_shape(self):
Expand Down