Fix lowvram dora issue.

2d416427 · comfyanonymous · ec6f16ad · 2d416427
Commit 2d416427 authored May 15, 2024 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/model_patcher.py comfy/model_patcher.py +1 -1

No files found.
--- 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()