From 2f0181405f25e1448a55697081e380020fe8c68d Mon Sep 17 00:00:00 2001
From: FNSpd <125805478+FNSpd@users.noreply.github.com>
Date: Tue, 21 Mar 2023 14:53:51 +0400
Subject: [PATCH] Update lora.py

---
 extensions-builtin/Lora/lora.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 8937b585e..7c371deb7 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -178,6 +178,7 @@ def load_loras(names, multipliers=None):
 
 
 def lora_forward(module, input, res):
+    input = devices.cond_cast_unet(input)
     if len(loaded_loras) == 0:
         return res