Skip to content

Commit d9029f2

Browse files
a-r-r-o-wsayakpaul
andauthored
[tests] fix name and unskip CogI2V integration test (#9683)
update Co-authored-by: Sayak Paul <[email protected]>
1 parent d204e53 commit d9029f2

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/pipelines/cogvideo/test_cogvideox_image2video.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
enable_full_determinism()
4444

4545

46-
class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
46+
class CogVideoXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
4747
pipeline_class = CogVideoXImageToVideoPipeline
4848
params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs"}
4949
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS.union({"image"})
@@ -343,7 +343,6 @@ def test_fused_qkv_projections(self):
343343
), "Original outputs should match when fused QKV projections are disabled."
344344

345345

346-
@unittest.skip("The model 'THUDM/CogVideoX-5b-I2V' is not public yet.")
347346
@slow
348347
@require_torch_gpu
349348
class CogVideoXImageToVideoPipelineIntegrationTests(unittest.TestCase):

0 commit comments

Comments
 (0)