Fix lowvram dora issue.

comfyanonymous · May 15, 2024 · 2d41642 · 2d41642
1 parent ec6f16a
commit 2d41642
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()