update

Author: a-r-r-o-w

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):