File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change 4343enable_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
349348class CogVideoXImageToVideoPipelineIntegrationTests (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments