Merge pull request #8782 from FNSpd/master

--upcast-sampling support for CUDA
2025-08-02 18:30:22 +00:00 · 2023-03-25 09:10:01 +03:00
parent b0b777e64d a9eab236d7
commit 03c8eefbcc
3 changed files with 4 additions and 3 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