diff --git a/extra/grpc/diffusers/backend_diffusers.py b/extra/grpc/diffusers/backend_diffusers.py index cff461f8..bb7d991a 100755 --- a/extra/grpc/diffusers/backend_diffusers.py +++ b/extra/grpc/diffusers/backend_diffusers.py @@ -300,6 +300,7 @@ class BackendServicer(backend_pb2_grpc.BackendServicer): "width": request.width, "height": request.height, "num_inference_steps": request.step, + "seed": request.seed, } if request.src != "": @@ -318,6 +319,12 @@ class BackendServicer(backend_pb2_grpc.BackendServicer): # create a dictionary of parameters by using the keys from EnableParameters and the values from defaults kwargs = {key: options[key] for key in keys} + # Set seed + if request.seed > 0: + kwargs["generator"] = torch.Generator(device="cuda").manual_seed( + request.seed + ) + image = {} if COMPEL: conditioning = self.compel.build_conditioning_tensor(prompt)