Update tests/pipelines/animatediff/test_animatediff_video2video.py

a-r-r-o-w · patrickvonplaten · web-flow · commit fdbb68f7d1b5 · 2024-01-19T17:02:28.000+05:30
Co-authored-by: Patrick von Platen &lt;patrick.v.platen@gmail.com&gt;
diff --git a/tests/pipelines/animatediff/test_animatediff_video2video.py b/tests/pipelines/animatediff/test_animatediff_video2video.py
@@ -38,8 +38,8 @@ class AnimateDiffVideoToVideoPipelineFastTests(PipelineTesterMixin, unittest.Tes
             "generator",
             "latents",
             "return_dict",
-            "callback",
-            "callback_steps",
+            "callback_on_step_end",
+            "callback_on_step_end_tensor_inputs",
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -38,8 +38,8 @@ class AnimateDiffVideoToVideoPipelineFastTests(PipelineTesterMixin, unittest.Tes`
`38`	`38`	`"generator",`
`39`	`39`	`"latents",`
`40`	`40`	`"return_dict",`
`41`		`- "callback",`
`42`		`- "callback_steps",`
	`41`	`+ "callback_on_step_end",`
	`42`	`+ "callback_on_step_end_tensor_inputs",`
`43`	`43`	`]`
`44`	`44`	`)`
`45`	`45`