Update app.py

app.py

@@ -24,11 +24,6 @@ def load_pipeline_w_lora() :
         revision=None
     )
 
-    # Load LoRa attn layer weights to unet attn layers
-    print('LoRa layers loading...')
-    unet.load_attn_procs(LoRa_DIR)
-    print('LoRa layers loaded')
-    
     # Load pipeline
     pipeline = DiffusionPipeline.from_pretrained(
         MODEL_REPO,
@@ -36,6 +31,12 @@ def load_pipeline_w_lora() :
         revision=None,
         torch_dtype=torch.float32,
     )
+    
+    # Load LoRa attn layer weights to unet attn layers
+    print('LoRa layers loading...')
+    pipeline.unet.load_attn_procs(LoRa_DIR)
+    print('LoRa layers loaded')
+    
     pipeline.set_progress_bar_config(disable=True)
 
     return pipeline