Update app.py

app.py

@@ -51,6 +51,18 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe.enable_model_cpu_offload()
 
+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
+
 
 #model_id = ("zer0int/LongCLIP-GmP-ViT-L-14")
 #config = CLIPConfig.from_pretrained(model_id)
@@ -155,14 +167,14 @@ css = '''
 #title img{width: 100px; margin-right: 0.5em}
 #gallery .grid-wrap{height: 10vh}
 '''
-with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
+with gr.Blocks(css=css) as app:
     title = gr.HTML(
         """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> SOONfactory </h1>""",
         elem_id="title",
     )
     	    # Info blob stating what the app is running
     info_blob = gr.HTML(
-        """<div id="info_blob"> Hosted Gallery of Custom-Trained Text2Image Generative Low-Rank Adaptors (LoRAs) for FLUX models. Running On: ArtsyLite FLUX base variant. First 4 gallery rows are adapters fine-tuned for the use of RCA (Revolutionary Communists of America at [https://CommunistUSA.org/]), & other activists/artists. Below them are adapters trained on works of Soviet Avant-Garde, Dada, Surrealism, & other radical styles + some original conceptions/fusions. Under those are identity models of notable revolutionaries & poets. Click squares to switch adapters & see links to their pages, many of them offering more info/resources. </div>"""
+        """<div id="info_blob"> Hosted Gallery of Custom-Trained Text2Image Generative Low-Rank Adaptors (LoRAs) for Z-image models. Running Over: Z.I.T. Originally set-up for adapters fine-tuned for the use of RCA (Revolutionary Communists of America at [https://CommunistUSA.org/]), & other activists/artists. We also train and feature adapters inspired by works of Soviet Avant-Garde, Dada, Surrealism, & other radical styles + some original conceptions/fusions. Under those are identity models of notable revolutionaries & poets. Click squares to switch adapters & see links to their pages, many of them offering more info/resources. </div>"""
     )
 
         # Info blob stating what the app is running