Update src/pipeline.py

src/pipeline.py

@@ -1,7 +1,7 @@
 import torch
 from PIL.Image import Image
 from diffusers import StableDiffusionXLPipeline
-from utils import register_parallel_pipeline_orig, register_faster_orig_forward
+# from utils import register_parallel_pipeline_orig, register_faster_orig_forward
 from pipelines.models import TextToImageRequest
 from diffusers import DDIMScheduler
 from torch import Generator
@@ -29,8 +29,8 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
 
     pipeline = compile_pipe(pipeline)
     load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
-    register_parallel_pipeline_orig(pipeline)
-    register_faster_orig_forward(pipeline)
+    # register_parallel_pipeline_orig(pipeline)
+    # register_faster_orig_forward(pipeline)
     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()