some cleanup

2024-11-09 15:15:10 +02:00
parent 643bbc18c1
commit 1c3aff9000
3 changed files with 14 additions and 34 deletions
@@ -816,7 +816,12 @@ class CogVideoSampler:
        base_path = pipeline["base_path"]

        assert "fun" not in base_path.lower(), "'Fun' models not supported in 'CogVideoSampler', use the 'CogVideoXFunSampler'"
-        assert ("I2V" not in pipeline.get("model_name","") or num_frames == 49 or context_options is not None), "I2V model can only do 49 frames"
+        assert (
+            "I2V" not in pipeline.get("model_name", "") or
+            "1.5" in pipeline.get("model_name", "") or
+            num_frames == 49 or
+            context_options is not None
+        ), "1.0 I2V model can only do 49 frames"

        device = mm.get_torch_device()
        offload_device = mm.unet_offload_device()