Use better dtype for the lowvram lora system.

6 months ago · 6138f92084
parent be0726c1ed
commit 6138f92084
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@ -96,7 +96,7 @@ class LowVramPatch:
        self.key = key
        self.model_patcher = model_patcher
    def __call__(self, weight):
-        return self.model_patcher.calculate_weight(self.model_patcher.patches[self.key], weight, self.key)
+        return self.model_patcher.calculate_weight(self.model_patcher.patches[self.key], weight, self.key, intermediate_dtype=weight.dtype)


 class ModelPatcher: