Fix lowvram dora issue.

6 months ago · 2d41642716
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@ -16,7 +16,7 @@ def apply_weight_decompose(dora_scale, weight):
        .transpose(0, 1)
    )

-    return weight * (dora_scale / weight_norm)
+    return weight * (dora_scale / weight_norm).type(weight.dtype)

 def set_model_options_patch_replace(model_options, patch, name, block_name, number, transformer_index=None):
    to = model_options["transformer_options"].copy()