From d200a54336305af76a76f8d06d38830cb2786707 Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Tue, 4 Mar 2025 21:39:18 +0100
Subject: [PATCH] fix crash if any lora

---
 gradio_server.py | 2 +-
 requirements.txt | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/gradio_server.py b/gradio_server.py
index e7a350d..fe6c2bb 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -350,7 +350,7 @@ def setup_loras(pipe,  lora_dir, lora_preselected_preset, split_linear_modules_m
 
     if len(loras) > 0:
         loras_names = [ Path(lora).stem for lora in loras  ]
-        offload.load_loras_into_model(pipe.transformer, loras,  activate_all_loras=False, split_linear_modules_map = split_linear_modules_map) #lora_multiplier,
+        offload.load_loras_into_model(pipe["transformer"], loras,  activate_all_loras=False, split_linear_modules_map = split_linear_modules_map) #lora_multiplier,
 
     if len(lora_preselected_preset) > 0:
         if not os.path.isfile(os.path.join(lora_dir, lora_preselected_preset + ".lset")):
diff --git a/requirements.txt b/requirements.txt
index 3d9a590..297f521 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,4 +16,5 @@ gradio>=5.0.0
 numpy>=1.23.5,<2
 einops
 moviepy==1.0.3
-mmgp==3.2.3
\ No newline at end of file
+mmgp==3.2.3
+peft==0.14.0
\ No newline at end of file