Update src/pipeline.py

src/pipeline.py

@@ -26,8 +26,8 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
         ).to("cuda")
 
     # Register optimizations for performance
-    register_parallel_pipeline(pipeline)
-    register_faster_forward(pipeline.unet)
+    # register_parallel_pipeline(pipeline)
+    # register_faster_forward(pipeline.unet)
     
     pipeline.scheduler = SchedulerWrapper(DDIMScheduler.from_config(pipeline.scheduler.config))
 
@@ -36,7 +36,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
 
     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