aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/Lora/lora.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-03-25 06:10:01 +0000
committerGitHub <noreply@github.com>2023-03-25 06:10:01 +0000
commit03c8eefbccaa054e3cc90cd43c14a701e71d1fe6 (patch)
tree48c2738abce1a20ca1637f244df6ba54e94d08b7 /extensions-builtin/Lora/lora.py
parentb0b777e64da238f1b259b77b0899b61c26a99dee (diff)
parenta9eab236d7e8afa4d6205127904a385b2c43bb24 (diff)
downloadstable-diffusion-webui-gfx803-03c8eefbccaa054e3cc90cd43c14a701e71d1fe6.tar.gz
stable-diffusion-webui-gfx803-03c8eefbccaa054e3cc90cd43c14a701e71d1fe6.tar.bz2
stable-diffusion-webui-gfx803-03c8eefbccaa054e3cc90cd43c14a701e71d1fe6.zip
Merge pull request #8782 from FNSpd/master
--upcast-sampling support for CUDA
Diffstat (limited to 'extensions-builtin/Lora/lora.py')
-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