Update src/pipeline.py

src/pipeline.py

@@ -39,14 +39,17 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
     
     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")
+    pipeline = compile_pipe(pipeline)
+    
+    # load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
 
-    for _ in range(1):
+    for _ in range(2):
         deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
     pipeline.scheduler.prepare_loss()
-    for _ in range(2):
+    for _ in range(4):
         pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
+    # save the compiled pipe
+    save_pipe(pipe, dir="/home/.cache/")
     return pipeline
 
 def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> Image: