diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-25 09:10:01 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-25 09:10:01 +0300 |
| commit | 03c8eefbccaa054e3cc90cd43c14a701e71d1fe6 (patch) | |
| tree | 48c2738abce1a20ca1637f244df6ba54e94d08b7 /extensions-builtin/Lora | |
| parent | b0b777e64da238f1b259b77b0899b61c26a99dee (diff) | |
| parent | a9eab236d7e8afa4d6205127904a385b2c43bb24 (diff) | |
Merge pull request #8782 from FNSpd/master
--upcast-sampling support for CUDA
Diffstat (limited to 'extensions-builtin/Lora')
| -rw-r--r-- | extensions-builtin/Lora/lora.py | 1 |
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
|