runner: None default for seed in pipelines

livepeer · Feb 2, 2024 · 75b7d29 · 75b7d29
1 parent 29ff04c
commit 75b7d29
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/runner/app/pipelines/image_to_image.py b/runner/app/pipelines/image_to_image.py
@@ -31,7 +31,7 @@ def __init__(self, model_id: str):
         self.ldm.to(get_torch_device())
 
     def __call__(self, prompt: str, image: PIL.Image, **kwargs) -> List[PIL.Image]:
-        seed = kwargs.pop("seed")
+        seed = kwargs.pop("seed", None)
         if seed is not None:
             kwargs["generator"] = torch.Generator(seed)
 

diff --git a/runner/app/pipelines/image_to_video.py b/runner/app/pipelines/image_to_video.py
@@ -34,7 +34,7 @@ def __call__(self, image: PIL.Image, **kwargs) -> List[List[PIL.Image]]:
         if "decode_chunk_size" not in kwargs:
             kwargs["decode_chunk_size"] = 8
 
-        seed = kwargs.pop("seed")
+        seed = kwargs.pop("seed", None)
         if seed is not None:
             kwargs["generator"] = torch.Generator(seed)
 

diff --git a/runner/app/pipelines/text_to_image.py b/runner/app/pipelines/text_to_image.py
@@ -31,7 +31,7 @@ def __init__(self, model_id: str):
         self.ldm.to(get_torch_device())
 
     def __call__(self, prompt: str, **kwargs) -> List[PIL.Image]:
-        seed = kwargs.pop("seed")
+        seed = kwargs.pop("seed", None)
         if seed is not None:
             kwargs["generator"] = torch.Generator(seed)