add

faaany · faaany · commit 28a73aca3cd9 · 2024-09-13T20:22:14.000-07:00
diff --git a/tests/pipelines/animatediff/test_animatediff.py b/tests/pipelines/animatediff/test_animatediff.py
@@ -300,8 +300,8 @@ def test_to_device(self):
         ]
         self.assertTrue(all(device == torch_device for device in model_devices))
 
-        output_cuda = pipe(**self.get_dummy_inputs(torch_device))[0]
-        self.assertTrue(np.isnan(to_np(output_cuda)).sum() == 0)
+        output_device = pipe(**self.get_dummy_inputs(torch_device))[0]
+        self.assertTrue(np.isnan(to_np(output_device)).sum() == 0)
 
     def test_to_dtype(self):
         components = self.get_dummy_components()

Original file line number	Diff line number	Diff line change
`@@ -300,8 +300,8 @@ def test_to_device(self):`
`300`	`300`	`]`
`301`	`301`	`self.assertTrue(all(device == torch_device for device in model_devices))`
`302`	`302`
`303`		`- output_cuda = pipe(**self.get_dummy_inputs(torch_device))[0]`
`304`		`- self.assertTrue(np.isnan(to_np(output_cuda)).sum() == 0)`
	`303`	`+ output_device = pipe(**self.get_dummy_inputs(torch_device))[0]`
	`304`	`+ self.assertTrue(np.isnan(to_np(output_device)).sum() == 0)`
`305`	`305`
`306`	`306`	`def test_to_dtype(self):`
`307`	`307`	`components = self.get_dummy_components()`