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 43
43
enable_full_determinism ()
44
44
45
45
46
- class CogVideoXPipelineFastTests (PipelineTesterMixin , unittest .TestCase ):
46
+ class CogVideoXImageToVideoPipelineFastTests (PipelineTesterMixin , unittest .TestCase ):
47
47
pipeline_class = CogVideoXImageToVideoPipeline
48
48
params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs" }
49
49
batch_params = TEXT_TO_IMAGE_BATCH_PARAMS .union ({"image" })
@@ -343,7 +343,6 @@ def test_fused_qkv_projections(self):
343
343
), "Original outputs should match when fused QKV projections are disabled."
344
344
345
345
346
- @unittest .skip ("The model 'THUDM/CogVideoX-5b-I2V' is not public yet." )
347
346
@slow
348
347
@require_torch_gpu
349
348
class CogVideoXImageToVideoPipelineIntegrationTests (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments