aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/Lora
diff options
context:
space:
mode:
authorFNSpd <125805478+FNSpd@users.noreply.github.com>2023-03-21 14:53:51 +0400
committerGitHub <noreply@github.com>2023-03-21 14:53:51 +0400
commit2f0181405f25e1448a55697081e380020fe8c68d (patch)
treed9e293ef34ba0cbb2bd034f4cbb79b76f12715e7 /extensions-builtin/Lora
parentc84c9df73799e173fcfafdc9548dbd043ba28682 (diff)
Update lora.py
Diffstat (limited to 'extensions-builtin/Lora')
-rw-r--r--extensions-builtin/Lora/lora.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 8937b585..7c371deb 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