Update src/pipeline.py

src/pipeline.py

@@ -1,6 +1,6 @@
 import torch
 from PIL.Image import Image
-from diffusers import UniPCMultistepScheduler, StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline
 from sfast.compilers.diffusion_pipeline_compiler import (compile,
                                                          CompilationConfig)
 from pipelines.models import TextToImageRequest
@@ -1252,7 +1252,7 @@ def load_pipeline() -> StableDiffusionXLPipeline:
         torch_dtype=torch.float16,
         local_files_only=True,
     )
-    pipeline.scheduler = UniPCMultistepScheduler.from_config('./src',)
+#    pipeline.scheduler = UniPCMultistepScheduler.from_config('./src',)
     pipeline.to("cuda")
     config = CompilationConfig.Default()
     try:
@@ -1286,5 +1286,5 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
         width=request.width,
         height=request.height,
         generator=generator,
-        num_inference_steps=11,
+        num_inference_steps=20,
     ).images[0]