Skip to content

Commit fdbb68f

Browse files
Update tests/pipelines/animatediff/test_animatediff_video2video.py
Co-authored-by: Patrick von Platen <[email protected]>
1 parent 54b21c0 commit fdbb68f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/animatediff/test_animatediff_video2video.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ class AnimateDiffVideoToVideoPipelineFastTests(PipelineTesterMixin, unittest.Tes
3838
"generator",
3939
"latents",
4040
"return_dict",
41-
"callback",
42-
"callback_steps",
41+
"callback_on_step_end",
42+
"callback_on_step_end_tensor_inputs",
4343
]
4444
)
4545

0 commit comments

Comments
 (0)