Update src/pipeline.py

src/pipeline.py

@@ -178,7 +178,7 @@ def load_pipeline() -> Pipeline:
     return pipeline
 
 @torch.no_grad()
-def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
+def infer(request: TextToImageRequest, pipeline: Pipeline, generator) -> Image:
     generator = Generator(pipeline.device).manual_seed(request.seed)
 
     return pipeline(