Unverified Commit 7e335319 authored by Anwesha Chowdhury's avatar Anwesha Chowdhury Committed by GitHub
Browse files

Fix deprecation warnings in test_ltx_image2video.py (#11538)



Fixed 2 warnings that were raised during running LTXImageToVideoPipelineFastTests
Co-authored-by: default avatarachowdhury1211@gmail.com <anwesha@LAPTOP-E5QGFMOQ>
parent 8c249d14
...@@ -109,7 +109,7 @@ class LTXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -109,7 +109,7 @@ class LTXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
else: else:
generator = torch.Generator(device=device).manual_seed(seed) generator = torch.Generator(device=device).manual_seed(seed)
image = torch.randn((1, 3, 32, 32), generator=generator, device=device) image = torch.rand((1, 3, 32, 32), generator=generator, device=device)
inputs = { inputs = {
"image": image, "image": image,
...@@ -142,7 +142,7 @@ class LTXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -142,7 +142,7 @@ class LTXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
self.assertEqual(generated_video.shape, (9, 3, 32, 32)) self.assertEqual(generated_video.shape, (9, 3, 32, 32))
expected_video = torch.randn(9, 3, 32, 32) expected_video = torch.randn(9, 3, 32, 32)
max_diff = np.abs(generated_video - expected_video).max() max_diff = torch.amax(torch.abs(generated_video - expected_video))
self.assertLessEqual(max_diff, 1e10) self.assertLessEqual(max_diff, 1e10)
def test_callback_inputs(self): def test_callback_inputs(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment