Support seed in diffusion models (#450)

* Support seed in diffusion models

add seed support

* black

* correct seed in generator

---------

Co-authored-by: multimodalart <[email protected]>

Author: apolinario

docker_images/diffusers/app/pipelines/text_to_image.py

@@ -169,5 +169,11 @@ def _process_req(self, inputs, **kwargs):
                 kwargs["num_inference_steps"] = 20
             # Else, don't specify anything, leave the default behaviour
 
+        if "seed" in kwargs:
+            seed = int(kwargs["seed"])
+            generator = torch.Generator().manual_seed(seed)
+            kwargs["generator"] = generator
+            kwargs.pop("seed")
+
         images = self.ldm(inputs, **kwargs)["images"]
         return images[0]