Update lora.py

2023-03-21 14:53:51 +04:00 · 2023-03-21 14:53:51 +04:00 · 2f0181405f
commit 2f0181405f
parent c84c9df737
1 changed files with 1 additions and 0 deletions
--- 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