Update pipeline.py

pipeline.py

@@ -505,6 +505,7 @@ class WebUIStableDiffusionPipeline(DiffusionPipeline):
         callback_steps: Optional[int] = 1,
         cross_attention_kwargs: Optional[Dict[str, Any]] = None,
         clip_skip: int = 1,
+        enable_lora: bool = True,
     ):
         r"""
         Function invoked when calling the pipeline for generation.
@@ -591,7 +592,7 @@ class WebUIStableDiffusionPipeline(DiffusionPipeline):
 
             prompts, extra_network_data = parse_prompts([prompt])
 
-            if self.LORA_DIR is not None:
+            if enable_lora and self.LORA_DIR is not None:
                 if os.path.exists(self.LORA_DIR):
                     lora_mapping = {p.stem: p.absolute() for p in Path(self.LORA_DIR).glob("*.safetensors")}
                     for params in extra_network_data["lora"]:
@@ -719,7 +720,7 @@ class WebUIStableDiffusionPipeline(DiffusionPipeline):
         except Exception as e:
             raise ValueError(e)
         finally:
-            if self.weights_has_changed:
+            if enable_lora and self.weights_has_changed:
                 for sub_layer in self.text_encoder.sublayers(include_self=True):
                     if hasattr(sub_layer, "backup_weights"):
                         sub_layer.weight.copy_(sub_layer.backup_weights, True)