diff --git a/tests/pipelines/cogvideo/test_cogvideox_image2video.py b/tests/pipelines/cogvideo/test_cogvideox_image2video.py index ec9a5fdd153e..f7e1fe7fd6c7 100644 --- a/tests/pipelines/cogvideo/test_cogvideox_image2video.py +++ b/tests/pipelines/cogvideo/test_cogvideox_image2video.py @@ -43,7 +43,7 @@ enable_full_determinism() -class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase): +class CogVideoXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase): pipeline_class = CogVideoXImageToVideoPipeline params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs"} batch_params = TEXT_TO_IMAGE_BATCH_PARAMS.union({"image"}) @@ -343,7 +343,6 @@ def test_fused_qkv_projections(self): ), "Original outputs should match when fused QKV projections are disabled." -@unittest.skip("The model 'THUDM/CogVideoX-5b-I2V' is not public yet.") @slow @require_torch_gpu class CogVideoXImageToVideoPipelineIntegrationTests(unittest.TestCase):