Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 import os
 from PIL import Image
 from diffusers import FluxKontextPipeline
-from diffusers.utils import load_image
+from diffusers.utils import load_image, peft_utils
 from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard
 from safetensors.torch import load_file
 import requests
@@ -18,6 +18,18 @@ MAX_SEED = np.iinfo(np.int32).max
 
 pipe = FluxKontextPipeline.from_pretrained("LPX55/FLUX.1_Kontext-Lightning", torch_dtype=torch.bfloat16).to("cuda")
 
+try: # A temp hack for some version diffusers lora loading problem
+    from diffusers.utils.peft_utils import _derive_exclude_modules
+
+    def new_derive_exclude_modules(*args, **kwargs):
+        exclude_modules = _derive_exclude_modules(*args, **kwargs)
+        if exclude_modules is not None:
+            exclude_modules = [n for n in exclude_modules if "proj_out" not in n]
+        return exclude_modules
+    peft_utils._derive_exclude_modules = new_derive_exclude_modules
+except:
+    pass
+
 # Load LoRA data (you'll need to create this JSON file or modify to load your LoRAs)
 
 with open("flux_loras.json", "r") as file: