Skip to content

GitLab

Unverified Commit d9029f2c authored by Aryan's avatar Aryan Committed by GitHub
Browse files

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



update
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent d204e532
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tests/pipelines/cogvideo/test_cogvideox_image2video.py
View file @ d9029f2c
......@@ -43,7 +43,7 @@ from ..test_pipelines_common import (
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 @@ class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
), "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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment