Skip to content

Commit

Permalink
Fix diffusers. From PR 1208 (huggingface#1210)
Browse files Browse the repository at this point in the history
  • Loading branch information
ssarkar2 authored Aug 6, 2024
1 parent 7c96602 commit 8ebd789
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_diffusers.py
Original file line number Diff line number Diff line change
Expand Up @@ -2436,7 +2436,7 @@ def test_stable_video_diffusion_single_video(self):
self.assertEqual(len(outputs), 1)
self.assertEqual(image.shape, (2, 3, 32, 32))

expected_slice = np.array([0.6285, 0.5813, 0.5434, 0.6494, 0.6302, 0.6262, 0.5426, 0.5404, 0.5203])
expected_slice = np.array([0.6208, 0.5780, 0.5447, 0.6462, 0.6285, 0.6288, 0.5334, 0.5287, 0.5165])

self.assertLess(np.abs(image_slice.flatten() - expected_slice).max(), 1e-2)

Expand Down

0 comments on commit 8ebd789

Please sign in to comment.