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
Show 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 ( ...@@ -43,7 +43,7 @@ from ..test_pipelines_common import (
enable_full_determinism() enable_full_determinism()
class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase): class CogVideoXImageToVideoPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
pipeline_class = CogVideoXImageToVideoPipeline pipeline_class = CogVideoXImageToVideoPipeline
params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs"} params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs"}
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS.union({"image"}) batch_params = TEXT_TO_IMAGE_BATCH_PARAMS.union({"image"})
...@@ -343,7 +343,6 @@ class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -343,7 +343,6 @@ class CogVideoXPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
), "Original outputs should match when fused QKV projections are disabled." ), "Original outputs should match when fused QKV projections are disabled."
@unittest.skip("The model 'THUDM/CogVideoX-5b-I2V' is not public yet.")
@slow @slow
@require_torch_gpu @require_torch_gpu
class CogVideoXImageToVideoPipelineIntegrationTests(unittest.TestCase): 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