Update src/pipeline.py

src/pipeline.py

@@ -1,6 +1,6 @@
 import torch
 from PIL.Image import Image
-from diffusers import UniPCMultistepScheduler, AutoencoderTiny
+from diffusers import UniPCMultistepScheduler, AutoencoderTiny, StableDiffusionXLPipeline
 from sfast.compilers.diffusion_pipeline_compiler import (compile,
                                                          CompilationConfig)
 from pipelines.models import TextToImageRequest
@@ -1290,7 +1290,10 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
 
 
 def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> Image:
-    generator = Generator(pipeline.device).manual_seed(request.seed) if request.seed else None
+    if request.seed is None:
+        generator = None
+    else:
+        generator = Generator(pipeline.device).manual_seed(request.seed)
 
     return pipeline(
         prompt=request.prompt,