Reset default scheduler when none is specified (#367)

* Reset default scheduler when none is specified
* revert also if incompatible

---------

Co-authored-by: Raphael Glon <[email protected]>
Co-authored-by: apolinário <[email protected]>

Author: oOraph

docker_images/diffusers/app/pipelines/text_to_image.py

@@ -85,6 +85,8 @@ def __init__(self, model_id: str):
                 self.ldm.scheduler.config
             )
 
+        self.default_scheduler = self.ldm.scheduler
+
         if not idle.UNLOAD_IDLE:
             self._model_to_gpu()
 
@@ -136,6 +138,11 @@ def __call__(self, inputs: str, **kwargs) -> "Image.Image":
                 self.ldm.scheduler = SchedulerClass.from_config(
                     self.ldm.scheduler.config
                 )
+            else:
+                logger.info("%s scheduler not loaded: incompatible", custom_scheduler)
+                self.ldm.scheduler = self.default_scheduler
+        else:
+            self.ldm.scheduler = self.default_scheduler
 
         self._load_lora_adapter(kwargs)