Remove incompatible params

livepeer · Apr 11, 2024 · 8d5120b · 8d5120b
1 parent b46a460
commit 8d5120b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/runner/app/pipelines/text_to_video.py b/runner/app/pipelines/text_to_video.py
@@ -53,6 +53,17 @@ def __call__(self, prompt: str, **kwargs) -> List[List[PIL.Image]]:
         if "decode_chunk_size" not in kwargs:
             kwargs["decode_chunk_size"] = 4
 
+        # ali-vilab/text-to-video-ms-1.7b has a limited parameter set
+        if (
+            self.model_id == "ali-vilab/text-to-video-ms-1.7b"
+        ):
+            if "fps" in kwargs:
+                del kwargs["fps"]
+            if "MotionBucketId" in kwargs:
+                del kwargs["MotionBucketId"]
+            if "NoiseAugStrength" in kwargs:
+                del kwargs["NoiseAugStrength"]
+
         seed = kwargs.pop("seed", None)
         if seed is not None:
             if isinstance(seed, int):