[tests] fix name and unskip CogI2V integration test (#9683)

update Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
huggingface · Oct 16, 2024 · d9029f2 · d9029f2
1 parent d204e53
commit d9029f2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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):