We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d204e53 commit d9029f2Copy full SHA for d9029f2
tests/pipelines/cogvideo/test_cogvideox_image2video.py
@@ -43,7 +43,7 @@
43
enable_full_determinism()
44
45
46
-class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
+class CogVideoXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
47
pipeline_class = CogVideoXImageToVideoPipeline
48
params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs"}
49
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS.union({"image"})
@@ -343,7 +343,6 @@ def test_fused_qkv_projections(self):
343
), "Original outputs should match when fused QKV projections are disabled."
344
345
346
-@unittest.skip("The model 'THUDM/CogVideoX-5b-I2V' is not public yet.")
347
@slow
348
@require_torch_gpu
349
class CogVideoXImageToVideoPipelineIntegrationTests(unittest.TestCase):
0 commit comments