Update src/pipeline.py

src/pipeline.py

@@ -25,14 +25,14 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
             torch_dtype=torch.float16,
         ).to("cuda")
     
-    # pipeline.scheduler = SchedulerWrapper(DDIMScheduler.from_config(pipeline.scheduler.config))
+    pipeline.scheduler = SchedulerWrapper(DDIMScheduler.from_config(pipeline.scheduler.config))
 
     pipeline = compile_pipe(pipeline)
     load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
 
     for _ in range(1):
         deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
-    # pipeline.scheduler.prepare_loss()
+    pipeline.scheduler.prepare_loss()
     for _ in range(2):
         pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
     return pipeline
@@ -49,7 +49,7 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
         width=request.width,
         height=request.height,
         generator=generator,
-        num_inference_steps=20,
+        num_inference_steps=13,
         cache_interval=1,
         cache_layer_id=1,
         cache_block_id=0,