diff --git a/composable_lora.py b/composable_lora.py
index 6258df1..93a039c 100644
--- a/composable_lora.py
+++ b/composable_lora.py
@@ -113,9 +113,13 @@ def lora_forward(compvis_module, input, res):
                     # tensor.shape[1] != uncond.shape[1]
                     cur_num_prompts = res.shape[0]
                     base = (diffusion_model_counter // cur_num_prompts) // num_loras * cur_num_prompts
-                    if 0 <= base < len(prompt_loras):
+                    prompt_len = len(prompt_loras)
+                    if 0 <= base < prompt_len:
                         # c
                         for off in range(cur_num_prompts):
+                            if base + off >= prompt_len:
+                                break
+                                
                             loras = prompt_loras[base + off]
                             multiplier = loras.get(lora.name, 0.0)
                             if multiplier != 0.0: