runner: Fix manual seeding for generators

Author: yondonfu

runner/app/pipelines/image_to_image.py

@@ -37,7 +37,7 @@ def __init__(self, model_id: str):
     def __call__(self, prompt: str, image: PIL.Image, **kwargs) -> List[PIL.Image]:
         seed = kwargs.pop("seed", None)
         if seed is not None:
-            kwargs["generator"] = torch.Generator(seed)
+            kwargs["generator"] = torch.Generator(get_torch_device()).manual_seed(seed)
 
         if (
             self.model_id == "stabilityai/sdxl-turbo"

runner/app/pipelines/image_to_video.py

@@ -40,7 +40,7 @@ def __call__(self, image: PIL.Image, **kwargs) -> List[List[PIL.Image]]:
 
         seed = kwargs.pop("seed", None)
         if seed is not None:
-            kwargs["generator"] = torch.Generator(seed)
+            kwargs["generator"] = torch.Generator(get_torch_device()).manual_seed(seed)
 
         return self.ldm(image, **kwargs).frames
 

runner/app/pipelines/text_to_image.py

@@ -33,7 +33,7 @@ def __init__(self, model_id: str):
     def __call__(self, prompt: str, **kwargs) -> List[PIL.Image]:
         seed = kwargs.pop("seed", None)
         if seed is not None:
-            kwargs["generator"] = torch.Generator(seed)
+            kwargs["generator"] = torch.Generator(get_torch_device()).manual_seed(seed)
 
         if (
             self.model_id == "stabilityai/sdxl-turbo"